Papers › A Bayesian generative neural network framework for epidemic inference problems

A Bayesian generative neural network framework for epidemic inference problems

5 Nov 2021arXiv:2111.03383archive 2025-07-28

Indaco Biazzo, Alfredo Braunstein, Luca Dall'Asta, Fabio Mazza

The reconstruction of missing information in epidemic spreading on contact networks can be essential in the prevention and containment strategies. The identification and warning of infectious but asymptomatic individuals (i.e., contact tracing), the well-known patient-zero problem, or the inference of the infectivity values in structured populations are examples of significant epidemic inference problems. As the number of possible epidemic cascades grows exponentially with the number of individuals involved and only an almost negligible subset of them is compatible with the observations (e.g., medical tests), epidemic inference in contact networks poses incredible computational challenges. We present a new generative neural networks framework that learns to generate the most probable infection cascades compatible with observations. The proposed method achieves better (in some cases, significantly better) or comparable results with existing methods in all problems considered both in synthetic and real contact networks. Given its generality, clear Bayesian and variational nature, the presented framework paves the way to solve fundamental inference epidemic problems with high precision in small and medium-sized real case scenarios such as the spread of infections in workplaces and hospitals.

PaperPDFCodeCode Syntology ran

In Syntology Open this paper in Syntology's Atlas, the map of the papers in Syntology's graph and their citations.

For agents, Syntology's MCP tool lists every function and class Syntology harvested from this paper and whether it ran (how to connect): get_harvested_code_for_paper(arxiv_id="2111.03383")

Code

Syntology Ran 0 of 24 code samples harvested from 1 repository linked to this paper; 24 have no recorded run.

By repository: official repository: 24 samples from 1 repository, 0 ran. The run record, sample by sample. “Ran” means executed on a synthesized input, not that the code is correct or reproduces the paper.

ocadni/annfore-results officialmentioned in papermentioned on GitHub report
ocadni/annfore officialmentioned on GitHubpytorchApache-2.0 report

Repository list and official/mentioned flags are the archive's, frozen 2025-07-28. Reachability, where shown, is from one Syntology probe window (2026-09-16 to 2026-09-18); repositories not probed show nothing. GitHub stars are not tracked.

Code Syntology ran Syntology

24 samples harvested; 0 ran; 0 honoured the contract we drafted; 24 have no recorded run. Read from Syntology's graph 2026-09-24; that is when this build read the record, not when the samples ran.

24unverified

Licence: 0 of the 24 samples are pointer only, meaning Syntology does not serve that copy's text. This page shows no code text for any sample; each one links to its file in the repository.

Harvested from ocadni/annfore. “Ran” means the sample executed on a synthesized input. It does not mean the output is correct, and nothing here reproduces the paper's results. “Honoured” and “violated” refer to a contract Syntology drafted from the code itself; “our draft was wrong” and “fixture could not drive it” are failures of Syntology's instrument, not of the code.

Each sample ends with its code_sha256, Syntology's identity for that exact code. An agent fetches the stored sample with Syntology's MCP tool get_code(code_sha256="…") (how to connect); click an identity to copy that call.

Repository labels, per sample. official repository: The archive marks this repository official for the paper. named in the paper: The archive records that the paper mentions this repository; it is not marked official. community (archive-listed): In the archive's code links for this paper, not marked official and not recorded as mentioned in the paper. found in paper text by Syntology: Syntology found this repository in the paper's own text; whether it is the authors' implementation is not asserted. community: Not in the archive's code links for this paper; a community repository Syntology harvested. Samples from a repository marked official are listed first. Licence labels name the repository's licence as recorded at harvest. “Pointer only” means Syntology does not serve that copy's text, for one of four reasons: no licence file was found; the licence was not identified; the licence is recorded as permissive but that copy's record is not marked cleared; or the licence is outside the permissive list Syntology serves text under (MIT, Apache-2.0, BSD and similar). Some licences outside that list permit redistribution, such as WTFPL, and GPL-3.0 under its conditions; they are simply not on the list. Hover a licence label for the reason. File links open the file on GitHub at the default branch, which may have changed since the harvest.

batch_mm ocadni/annfore/annfore/models/common.py official repository unverified Apache-2.0 (permissive) · 88a25afdb229a4de · report
calc_avg_states ocadni/annfore/annfore/learn/l_utils.py official repository unverified Apache-2.0 (permissive) · e46492b8d6cb0304 · report
calc_feat_power ocadni/annfore/annfore/net/deep_linear.py official repository unverified Apache-2.0 (permissive) · f32d4ccb471ff352 · report
calc_p_no_c_np ocadni/annfore/annfore/models/common.py official repository unverified Apache-2.0 (permissive) · d30029f478a818ae · report
check_convert_tensor ocadni/annfore/annfore/models/common.py official repository unverified Apache-2.0 (permissive) · 169404d17a6af306 · report
check_convert_tensor ocadni/annfore/annfore/models/sir_model_sep.py official repository unverified Apache-2.0 (permissive) · 7010edca19a563be · report
format_src_print ocadni/annfore/annfore/learn/l_utils.py official repository unverified Apache-2.0 (permissive) · 8e6aaa40cc72bcbb · report
get_linear_layers_path ocadni/annfore/annfore/net/base.py official repository unverified Apache-2.0 (permissive) · 5e93c65f419d5914 · report
logP ocadni/annfore/annfore/learn/train_params.py official repository unverified Apache-2.0 (permissive) · db96eb179ca980fe · report
logP_logQ ocadni/annfore/annfore/learn/train_params.py official repository unverified Apache-2.0 (permissive) · 33352cf70ab668d7 · report
loss_fn ocadni/annfore/annfore/learn/losses.py official repository unverified Apache-2.0 (permissive) · 8260a897032b015e · report
loss_fn_coeff ocadni/annfore/annfore/learn/losses.py official repository unverified Apache-2.0 (permissive) · a7f1d520148c4025 · report
loss_fn_coeff_p_sus ocadni/annfore/annfore/learn/losses.py official repository unverified Apache-2.0 (permissive) · 22fae26097751c91 · report
make_lin_layers ocadni/annfore/annfore/net/deep_linear.py official repository unverified Apache-2.0 (permissive) · 70402e4f401c4103 · report
make_masks ocadni/annfore/annfore/net/nn_sir_path_obs.py official repository unverified Apache-2.0 (permissive) · 88747e21714a2d36 · report
make_opt ocadni/annfore/annfore/learn/opt.py official repository unverified Apache-2.0 (permissive) · f4f47a0dd89c33fc · report
my_linear ocadni/annfore/annfore/net/deep_linear.py official repository unverified Apache-2.0 (permissive) · 30a691fc43d7825e · report
normalize ocadni/annfore/annfore/models/sir_model_sep.py official repository unverified Apache-2.0 (permissive) · 90ce614d4b6b5793 · report
one_hot_conf_from_times ocadni/annfore/annfore/utils/conf_utils.py official repository unverified Apache-2.0 (permissive) · 23f65953c94eee38 · report
one_hot_samples ocadni/annfore/annfore/utils/conf_utils.py official repository unverified Apache-2.0 (permissive) · 3eb2444e97ae7693 · report
one_hot_samples_relative_r ocadni/annfore/annfore/utils/conf_utils.py official repository unverified Apache-2.0 (permissive) · b326d2cdb16132fc · report
opt_param_init ocadni/annfore/annfore/learn/train_params.py official repository unverified Apache-2.0 (permissive) · 25fc7691e23186cf · report
sample_avg_states ocadni/annfore/annfore/learn/l_utils.py official repository unverified Apache-2.0 (permissive) · 8de71585e15ab1b8 · report
transform_to_spins ocadni/annfore/annfore/models/sir_model_sep.py official repository unverified Apache-2.0 (permissive) · 9fb5bf91b56bc932 · report

Results from the paper archive 2025-07-28

No leaderboard rows for this paper in the archive.

Report a problem or propose a change · a person checks every report against the paper or source before anything changes; decisions are listed on /corrections