{"about":{"site":"https://codewithpapers.app","non_affiliation":"Code with Papers and Syntology are not affiliated with, endorsed by, or sponsored by Papers with Code, Meta, or the pwc-archive mirror.","licence":"CC BY-SA 4.0","licence_url":"https://creativecommons.org/licenses/by-sa/4.0/legalcode","attribution":"https://codewithpapers.app/attribution","modified":"archive material modified by Syntology; see the attribution page"},"url":"/paper/arxiv-2602-05286","title":"HealthMamba: An Uncertainty-aware Spatiotemporal Graph State Space Model for Effective and Reliable Healthcare Facility Visit Prediction","arxiv_id":"2602.05286","date":"2026-02-05","proceeding":null,"authors":["Dahai Yu","Lin Jiang","Rongchao Xu","Guang Wang"],"abstract":"Healthcare facility visit prediction is essential for optimizing healthcare resource allocation and informing public health policy. Despite advanced machine learning methods being employed for better prediction performance, existing works usually formulate this task as a time-series forecasting problem without considering the intrinsic spatial dependencies of different types of healthcare facilities, and they also fail to provide reliable predictions under abnormal situations such as public emergencies. To advance existing research, we propose HealthMamba, an uncertainty-aware spatiotemporal framework for accurate and reliable healthcare facility visit prediction. HealthMamba comprises three key components: (i) a Unified Spatiotemporal Context Encoder that fuses heterogeneous static and dynamic information, (ii) a novel Graph State Space Model called GraphMamba for hierarchical spatiotemporal modeling, and (iii) a comprehensive uncertainty quantification module integrating three uncertainty quantification mechanisms for reliable prediction. We evaluate HealthMamba on four large-scale real-world datasets from California, New York, Texas, and Florida. Results show HealthMamba achieves around 6.0% improvement in prediction accuracy and 3.5% improvement in uncertainty quantification over state-of-the-art baselines.","url_abs":"https://arxiv.org/abs/2602.05286","url_pdf":"https://arxiv.org/pdf/2602.05286","source":{"archive":null,"snapshot":"2025-07-28","note":"not in the Papers with Code archive (frozen at the snapshot)","row_kind":"graph","title_abstract_authors_date":"arXiv metadata, CC0 1.0 (https://info.arxiv.org/help/license)"},"code_links":[],"tasks":[],"methods":[],"datasets_introduced":[],"methods_introduced":[],"results":[],"syntology":{"atlas_url":"https://app.syntology.ai/?focus=2602.05286","mcp":{"get_harvested_code_for_paper":{"arxiv_id":"2602.05286"}},"developers":"https://syntology.ai/developers","read_at":"2026-09-24T18:15:14+00:00","read_at_is":"when the build read Syntology's graph, not when any sample ran","claim":"Per-sample execution status on synthesized fixtures; not a correctness claim about the paper. Samples come from repositories linked to the paper, official or community; repo_kind says which.","repos":[{"mentioned_in_github":null,"is_official":null,"provenance":"deterministic:regex_extraction","mentioned_in_paper":null,"url":"https://github.com/UFOdestiny/HealthMamba","reach":null}],"summary":{"unverified":9},"by_repo_kind":{"found_in_text":{"samples":9,"ran":0,"repositories":1}},"repo_kind_vocabulary":{"official":"The archive marks this repository official for the paper","named_in_paper":"The archive records that the paper mentions this repository; it is not marked official","listed":"In the archive's code links for this paper, not marked official and not recorded as mentioned in the paper","found_in_text":"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"},"n_pointer_only_for_licence":0,"samples":[{"code_sha256_prefix":"2976735704f9e258","entry":"calculate_KGCN","repo":"UFOdestiny/HealthMamba","repo_kind":"found_in_text","path":"utils/graph_algo.py","file_url":"https://github.com/UFOdestiny/HealthMamba/blob/HEAD/utils/graph_algo.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"2976735704f9e258"}},{"code_sha256_prefix":"a616b4dba35c9611","entry":"calculate_normalized_laplacian","repo":"UFOdestiny/HealthMamba","repo_kind":"found_in_text","path":"utils/graph_algo.py","file_url":"https://github.com/UFOdestiny/HealthMamba/blob/HEAD/utils/graph_algo.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"a616b4dba35c9611"}},{"code_sha256_prefix":"9c429a6050647bb1","entry":"get_log_path","repo":"UFOdestiny/HealthMamba","repo_kind":"found_in_text","path":"utils/args.py","file_url":"https://github.com/UFOdestiny/HealthMamba/blob/HEAD/utils/args.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"9c429a6050647bb1"}},{"code_sha256_prefix":"e4153bb6ddac9c81","entry":"get_mask","repo":"UFOdestiny/HealthMamba","repo_kind":"found_in_text","path":"base/metrics.py","file_url":"https://github.com/UFOdestiny/HealthMamba/blob/HEAD/base/metrics.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"e4153bb6ddac9c81"}},{"code_sha256_prefix":"57c2470631d2c4d2","entry":"masked_mae","repo":"UFOdestiny/HealthMamba","repo_kind":"found_in_text","path":"base/metrics.py","file_url":"https://github.com/UFOdestiny/HealthMamba/blob/HEAD/base/metrics.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"57c2470631d2c4d2"}},{"code_sha256_prefix":"64aac06518bc2b5b","entry":"masked_mse","repo":"UFOdestiny/HealthMamba","repo_kind":"found_in_text","path":"base/metrics.py","file_url":"https://github.com/UFOdestiny/HealthMamba/blob/HEAD/base/metrics.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"64aac06518bc2b5b"}},{"code_sha256_prefix":"23bd6fab270f733e","entry":"normalize_adj_mx","repo":"UFOdestiny/HealthMamba","repo_kind":"found_in_text","path":"utils/graph_algo.py","file_url":"https://github.com/UFOdestiny/HealthMamba/blob/HEAD/utils/graph_algo.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"23bd6fab270f733e"}},{"code_sha256_prefix":"1020df1b2805a3fe","entry":"resolve_engine_template","repo":"UFOdestiny/HealthMamba","repo_kind":"found_in_text","path":"utils/args.py","file_url":"https://github.com/UFOdestiny/HealthMamba/blob/HEAD/utils/args.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"1020df1b2805a3fe"}},{"code_sha256_prefix":"b6a0b77056c5bad6","entry":"tuple_type","repo":"UFOdestiny/HealthMamba","repo_kind":"found_in_text","path":"utils/args.py","file_url":"https://github.com/UFOdestiny/HealthMamba/blob/HEAD/utils/args.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"b6a0b77056c5bad6"}}]},"arxiv_metadata":{"licence":"arXiv metadata, CC0 1.0 (https://info.arxiv.org/help/license)","fields":["title","abstract","authors","date"],"primary_category":"cs.LG","source":"arxiv_2026.jsonl"},"syntology_extracted_results":null}