{"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-2509-19885","title":"Towards Self-Supervised Foundation Models for Critical Care Time Series","arxiv_id":"2509.19885","date":"2025-09-24","proceeding":"NeurIPS","authors":["Katja Naasunnguaq Jagd","Rachael DeVries","Ole Winther"],"abstract":"Domain-specific foundation models for healthcare have expanded rapidly in recent years, yet foundation models for critical care time series remain relatively underexplored due to the limited size and availability of datasets. In this work, we introduce an early-stage pre-trained foundation model for critical care time-series based on the Bi-Axial Transformer (BAT), trained on pooled electronic health record datasets. We demonstrate effective transfer learning by fine-tuning the model on a dataset distinct from the training sources for mortality prediction, where it outperforms supervised baselines, particularly for small datasets ($<5,000$). These contributions highlight the potential of self-supervised foundation models for critical care times series to support generalizable and robust clinical applications in resource-limited settings.","url_abs":"https://arxiv.org/abs/2509.19885","url_pdf":"https://arxiv.org/pdf/2509.19885","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":{"syntology_url":null,"atlas_url":"https://app.syntology.ai/?focus=2509.19885","mcp":{"get_harvested_code_for_paper":{"arxiv_id":"2509.19885"}},"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/Katja-Jagd/YAIB","reach":{"status":"ok","spdx":"MIT"}},{"mentioned_in_github":null,"is_official":null,"provenance":"deterministic:regex_extraction","mentioned_in_paper":null,"url":"https://github.com/rvandewater/YAIB","reach":{"status":"ok","spdx":"MIT"}}],"summary":{"ran_draft_wrong":3,"unverified":21},"by_repo_kind":{"found_in_text":{"samples":24,"ran":3,"repositories":2}},"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":"0f786c407fb1ee4c","entry":"swish","repo":"rvandewater/YAIB","repo_kind":"found_in_text","path":"icu_benchmarks/imputation/diffwave.py","file_url":"https://github.com/rvandewater/YAIB/blob/HEAD/icu_benchmarks/imputation/diffwave.py","link_basis":"harvester_set","language":"python","status":"ran_draft_wrong","verification_level":2,"contract_check":"MISDECLARED","metamorphic_tier":"invariant","behaviour_fingerprint":true,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"0f786c407fb1ee4c"}},{"code_sha256_prefix":"7340b482ffe44836","entry":"Conv1d_with_init","repo":"rvandewater/YAIB","repo_kind":"found_in_text","path":"icu_benchmarks/imputation/csdi.py","file_url":"https://github.com/rvandewater/YAIB/blob/HEAD/icu_benchmarks/imputation/csdi.py","link_basis":"plan_row","language":"python","status":"ran_draft_wrong","verification_level":1,"contract_check":"OUTPUT_MISDECLARED","metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"7340b482ffe44836"}},{"code_sha256_prefix":"e706793d7d22ae44","entry":"get_torch_trans","repo":"rvandewater/YAIB","repo_kind":"found_in_text","path":"icu_benchmarks/imputation/csdi.py","file_url":"https://github.com/rvandewater/YAIB/blob/HEAD/icu_benchmarks/imputation/csdi.py","link_basis":"plan_row","language":"python","status":"ran_draft_wrong","verification_level":1,"contract_check":"OUTPUT_MISDECLARED","metamorphic_tier":"deterministic","behaviour_fingerprint":false,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"e706793d7d22ae44"}},{"code_sha256_prefix":"6e037d50ada32ff6","entry":"BO_mask","repo":"rvandewater/YAIB","repo_kind":"found_in_text","path":"icu_benchmarks/imputation/amputations.py","file_url":"https://github.com/rvandewater/YAIB/blob/HEAD/icu_benchmarks/imputation/amputations.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":"6e037d50ada32ff6"}},{"code_sha256_prefix":"23a54dd969a6cc4e","entry":"MAR_logistic_mask","repo":"rvandewater/YAIB","repo_kind":"found_in_text","path":"icu_benchmarks/imputation/amputations.py","file_url":"https://github.com/rvandewater/YAIB/blob/HEAD/icu_benchmarks/imputation/amputations.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":"23a54dd969a6cc4e"}},{"code_sha256_prefix":"c9b8608abd0c8fda","entry":"MCAR_mask","repo":"rvandewater/YAIB","repo_kind":"found_in_text","path":"icu_benchmarks/imputation/amputations.py","file_url":"https://github.com/rvandewater/YAIB/blob/HEAD/icu_benchmarks/imputation/amputations.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":"c9b8608abd0c8fda"}},{"code_sha256_prefix":"913d82af53065418","entry":"accuracy","repo":"rvandewater/YAIB","repo_kind":"found_in_text","path":"icu_benchmarks/models/custom_metrics.py","file_url":"https://github.com/rvandewater/YAIB/blob/HEAD/icu_benchmarks/models/custom_metrics.py","link_basis":"plan_row","language":"python","status":"unverified","verification_level":0,"contract_check":"RAISES","metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"913d82af53065418"}},{"code_sha256_prefix":"0741c6a2d1c22012","entry":"apply_wandb_sweep","repo":"rvandewater/YAIB","repo_kind":"found_in_text","path":"icu_benchmarks/wandb_utils.py","file_url":"https://github.com/rvandewater/YAIB/blob/HEAD/icu_benchmarks/wandb_utils.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":"0741c6a2d1c22012"}},{"code_sha256_prefix":"38373950dd739e60","entry":"calc_diffusion_hyperparams","repo":"rvandewater/YAIB","repo_kind":"found_in_text","path":"icu_benchmarks/imputation/diffwave.py","file_url":"https://github.com/rvandewater/YAIB/blob/HEAD/icu_benchmarks/imputation/diffwave.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":"38373950dd739e60"}},{"code_sha256_prefix":"5e384ebd2c7aa5ca","entry":"calc_diffusion_step_embedding","repo":"rvandewater/YAIB","repo_kind":"found_in_text","path":"icu_benchmarks/imputation/diffwave.py","file_url":"https://github.com/rvandewater/YAIB/blob/HEAD/icu_benchmarks/imputation/diffwave.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":"5e384ebd2c7aa5ca"}},{"code_sha256_prefix":"92336a4643e58418","entry":"confusion_matrix","repo":"rvandewater/YAIB","repo_kind":"found_in_text","path":"icu_benchmarks/models/custom_metrics.py","file_url":"https://github.com/rvandewater/YAIB/blob/HEAD/icu_benchmarks/models/custom_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":"92336a4643e58418"}},{"code_sha256_prefix":"24a54df4893018f7","entry":"create_optimizer","repo":"rvandewater/YAIB","repo_kind":"found_in_text","path":"icu_benchmarks/models/utils.py","file_url":"https://github.com/rvandewater/YAIB/blob/HEAD/icu_benchmarks/models/utils.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":"24a54df4893018f7"}},{"code_sha256_prefix":"a5e1412c8a01d50a","entry":"create_run_dir","repo":"rvandewater/YAIB","repo_kind":"found_in_text","path":"icu_benchmarks/run_utils.py","file_url":"https://github.com/rvandewater/YAIB/blob/HEAD/icu_benchmarks/run_utils.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":"a5e1412c8a01d50a"}},{"code_sha256_prefix":"270211d5760b3ffc","entry":"create_scheduler","repo":"rvandewater/YAIB","repo_kind":"found_in_text","path":"icu_benchmarks/models/utils.py","file_url":"https://github.com/rvandewater/YAIB/blob/HEAD/icu_benchmarks/models/utils.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":"270211d5760b3ffc"}},{"code_sha256_prefix":"990c19c698666ac3","entry":"derive_obs_mask_from_sensor_mask","repo":"Katja-Jagd/YAIB","repo_kind":"found_in_text","path":"icu_benchmarks/fine_tuning_classification_per_timestep.py","file_url":"https://github.com/Katja-Jagd/YAIB/blob/HEAD/icu_benchmarks/fine_tuning_classification_per_timestep.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":"990c19c698666ac3"}},{"code_sha256_prefix":"3141c06763955c96","entry":"eval_constant_predictor","repo":"Katja-Jagd/YAIB","repo_kind":"found_in_text","path":"icu_benchmarks/fine_tuning_regression_per_timestep.py","file_url":"https://github.com/Katja-Jagd/YAIB/blob/HEAD/icu_benchmarks/fine_tuning_regression_per_timestep.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":"3141c06763955c96"}},{"code_sha256_prefix":"ff33ca62f52a82c6","entry":"fmt","repo":"Katja-Jagd/YAIB","repo_kind":"found_in_text","path":"finetuning_results/pretrained_BAT/summarize_results.py","file_url":"https://github.com/Katja-Jagd/YAIB/blob/HEAD/finetuning_results/pretrained_BAT/summarize_results.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":"ff33ca62f52a82c6"}},{"code_sha256_prefix":"20a234617c6040ca","entry":"get_config_files","repo":"rvandewater/YAIB","repo_kind":"found_in_text","path":"icu_benchmarks/run_utils.py","file_url":"https://github.com/rvandewater/YAIB/blob/HEAD/icu_benchmarks/run_utils.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":"20a234617c6040ca"}},{"code_sha256_prefix":"c3894948f22f015a","entry":"load_pretrained_imputation_model","repo":"rvandewater/YAIB","repo_kind":"found_in_text","path":"icu_benchmarks/run_utils.py","file_url":"https://github.com/rvandewater/YAIB/blob/HEAD/icu_benchmarks/run_utils.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":"c3894948f22f015a"}},{"code_sha256_prefix":"8b5606c569acf0f3","entry":"load_subset","repo":"Katja-Jagd/YAIB","repo_kind":"found_in_text","path":"icu_benchmarks/fine_tuning_utils.py","file_url":"https://github.com/Katja-Jagd/YAIB/blob/HEAD/icu_benchmarks/fine_tuning_utils.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":"8b5606c569acf0f3"}},{"code_sha256_prefix":"70accb30753b78c5","entry":"load_subset_as_data_dict","repo":"Katja-Jagd/YAIB","repo_kind":"found_in_text","path":"icu_benchmarks/fine_tuning_utils.py","file_url":"https://github.com/Katja-Jagd/YAIB/blob/HEAD/icu_benchmarks/fine_tuning_utils.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":"70accb30753b78c5"}},{"code_sha256_prefix":"99e5e558884ea88c","entry":"maybe_extract_obs_mask","repo":"Katja-Jagd/YAIB","repo_kind":"found_in_text","path":"icu_benchmarks/fine_tuning_classification_per_timestep.py","file_url":"https://github.com/Katja-Jagd/YAIB/blob/HEAD/icu_benchmarks/fine_tuning_classification_per_timestep.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":"99e5e558884ea88c"}},{"code_sha256_prefix":"71f1f25db46d6880","entry":"parse_log_file","repo":"Katja-Jagd/YAIB","repo_kind":"found_in_text","path":"finetuning_results/extract_results_out_file.py","file_url":"https://github.com/Katja-Jagd/YAIB/blob/HEAD/finetuning_results/extract_results_out_file.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":"71f1f25db46d6880"}},{"code_sha256_prefix":"0c1ba10d581571e0","entry":"state_dict_transfer_report","repo":"Katja-Jagd/YAIB","repo_kind":"found_in_text","path":"icu_benchmarks/fine_tuning_regression_per_timestep.py","file_url":"https://github.com/Katja-Jagd/YAIB/blob/HEAD/icu_benchmarks/fine_tuning_regression_per_timestep.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":"0c1ba10d581571e0"}}]},"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_api"},"syntology_extracted_results":null}