← Search

Jiaming Cui

10 accepted papers

2026

How (Not) to Hybridize Neural and Mechanistic Models for Epidemiological Forecasting

ICML 2026poster

Epidemiological forecasting from surveillance data is a hard problem and hybridizing mechanistic compartmental models with neural models is a natural direction. The mechanistic structure helps keep trajectories epidemiologically plausible, while neural components can capture non-stationary, data-ada…

Cited by 0SourceScholar
2025

DF$^2$: Distribution-Free Decision-Focused Learning

UAI 2025

Decision-focused learning (DFL), which differentiates through the KKT conditions, has recently emerged as a powerful approach for predict-then-optimize problems. However, under probabilistic settings, DFL faces three major bottlenecks: model mismatch error, sample average approximation error, and gr

2024

Time-MMD: Multi-Domain Multimodal Dataset for Time Series Analysis

NeurIPS 2024poster

Time series data are ubiquitous across a wide range of real-world domains. While real-world time series analysis (TSA) requires human experts to integrate numerical series data with multimodal domain-specific knowledge, most existing TSA models rely solely on numerical data, overlooking the signific…

2023

Autoregressive Diffusion Model for Graph Generation

ICML 2023poster

Diffusion-based graph generative models have recently obtained promising results for graph generation. However, existing diffusion-based graph generative models are mostly one-shot generative models that apply Gaussian diffusion in the dequantized adjacency matrix space. Such a strategy can suffer f…

Cited by 71SourcePDFScholar
2023

Detecting Sources of Healthcare Associated Infections

AAAI 2023technical

Healthcare acquired infections (HAIs) (e.g., Methicillin-resistant Staphylococcus aureus infection) have complex transmission pathways, spreading not just via direct person-to-person contacts, but also via contaminated surfaces. Prior work in mathematical epidemiology has led to a class of models –…

2023

EINNs: Epidemiologically-Informed Neural Networks

AAAI 2023technical

We introduce EINNs, a framework crafted for epidemic forecasting that builds upon the theoretical grounds provided by mechanistic models as well as the data-driven expressibility afforded by AI models, and their capabilities to ingest heterogeneous information. Although neural forecasting models hav…

2022

End-to-end Stochastic Optimization with Energy-based Model

NeurIPS 2022accept

Decision-focused learning (DFL) was recently proposed for stochastic optimization problems that involve unknown parameters. By integrating predictive modeling with an implicitly differentiable optimization layer, DFL has shown superior performance to the standard two-stage predict-then-optimize pipe…

2022

Provable Sensor Sets for Epidemic Detection over Networks with Minimum Delay

AAAI 2022technical

The efficient detection of outbreaks and other cascading phenomena is a fundamental problem in a number of domains, including disease spread, social networks, and infrastructure networks. In such settings, monitoring and testing a small group of pre-selected nodes from the susceptible population (i.…

Cited by 4SourcePDFScholar