{"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/beds-bench-behavior-of-ehr-models-under","title":"BEDS-Bench: Behavior of EHR-models under Distributional Shift--A Benchmark","arxiv_id":"2107.08189","date":"2021-07-17","proceeding":null,"authors":["Anand Avati","Martin Seneviratne","Emily Xue","Zhen Xu","Balaji Lakshminarayanan","Andrew M. Dai"],"abstract":"Machine learning has recently demonstrated impressive progress in predictive accuracy across a wide array of tasks. Most ML approaches focus on generalization performance on unseen data that are similar to the training data (In-Distribution, or IND). However, real world applications and deployments of ML rarely enjoy the comfort of encountering examples that are always IND. In such situations, most ML models commonly display erratic behavior on Out-of-Distribution (OOD) examples, such as assigning high confidence to wrong predictions, or vice-versa. Implications of such unusual model behavior are further exacerbated in the healthcare setting, where patient health can potentially be put at risk. It is crucial to study the behavior and robustness properties of models under distributional shift, understand common failure modes, and take mitigation steps before the model is deployed. Having a benchmark that shines light upon these aspects of a model is a first and necessary step in addressing the issue. Recent work and interest in increasing model robustness in OOD settings have focused more on image modality, while the Electronic Health Record (EHR) modality is still largely under-explored. We aim to bridge this gap by releasing BEDS-Bench, a benchmark for quantifying the behavior of ML models over EHR data under OOD settings. We use two open access, de-identified EHR datasets to construct several OOD data settings to run tests on, and measure relevant metrics that characterize crucial aspects of a model's OOD behavior. We evaluate several learning algorithms under BEDS-Bench and find that all of them show poor generalization performance under distributional shift in general. Our results highlight the need and the potential to improve robustness of EHR models under distributional shift, and BEDS-Bench provides one way to measure progress towards that goal.","url_abs":"https://arxiv.org/abs/2107.08189v1","url_pdf":"https://arxiv.org/pdf/2107.08189v1.pdf","source":{"archive":"pwc-archive (Hugging Face), CC BY-SA 4.0","snapshot":"2025-07-28","licence_url":"https://creativecommons.org/licenses/by-sa/4.0/legalcode","row_kind":"abstracts"},"code_links":[{"paper_slug":"beds-bench-behavior-of-ehr-models-under","repo_url":"https://github.com/google-health/records-research","is_official":1,"mentioned_in_paper":1,"mentioned_in_github":0,"framework":"tf","reach":{"status":"ok"}}],"tasks":[],"methods":[],"datasets_introduced":[],"methods_introduced":[],"results":[],"syntology":{"syntology_url":"https://syntology.ai/paper/2107.08189","atlas_url":"https://app.syntology.ai/?focus=2107.08189","mcp":{"get_harvested_code_for_paper":{"arxiv_id":"2107.08189"}},"developers":"https://syntology.ai/developers","read_at":"2026-09-25T09:33:49+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":[{"provenance":"external:paperswithcode_snapshot_2025-07-28","url":"https://github.com/google-health/records-research","reach":{"status":"ok"}}],"summary":{"ran_fixture":2,"ran_draft_wrong":1,"ran_honours":2,"unverified":1},"by_repo_kind":{"official":{"samples":6,"ran":5,"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":6,"samples":[{"code_sha256_prefix":"9e87d7e7bf3bc133","entry":"config_get","repo":"google-health/records-research","repo_kind":"official","path":"beds-bench/benchmark.py","file_url":"https://github.com/google-health/records-research/blob/HEAD/beds-bench/benchmark.py","link_basis":"first_harvest_node","language":"python","status":"ran_fixture","verification_level":1,"contract_check":"RAISES","metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"9e87d7e7bf3bc133"}},{"code_sha256_prefix":"a16ad595e3803161","entry":"fit","repo":"google-health/records-research","repo_kind":"official","path":"beds-bench/models/GP.py","file_url":"https://github.com/google-health/records-research/blob/HEAD/beds-bench/models/GP.py","link_basis":"first_harvest_node","language":"python","status":"ran_fixture","verification_level":1,"contract_check":"RAISES","metamorphic_tier":"deterministic","behaviour_fingerprint":false,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"a16ad595e3803161"}},{"code_sha256_prefix":"a625fb2efe302935","entry":"fit_model","repo":"google-health/records-research","repo_kind":"official","path":"beds-bench/benchmark.py","file_url":"https://github.com/google-health/records-research/blob/HEAD/beds-bench/benchmark.py","link_basis":"first_harvest_node","language":"python","status":"ran_draft_wrong","verification_level":1,"contract_check":"MISDECLARED","metamorphic_tier":"deterministic","behaviour_fingerprint":false,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"a625fb2efe302935"}},{"code_sha256_prefix":"07fc687cca9b8e69","entry":"had2enc","repo":"google-health/records-research","repo_kind":"official","path":"beds-bench/prepare.py","file_url":"https://github.com/google-health/records-research/blob/HEAD/beds-bench/prepare.py","link_basis":"first_harvest_node","language":"python","status":"ran_honours","verification_level":1,"contract_check":"HONOURS","metamorphic_tier":"well_formed","behaviour_fingerprint":true,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"07fc687cca9b8e69"}},{"code_sha256_prefix":"cddb6efa0ce3d290","entry":"sub2pid","repo":"google-health/records-research","repo_kind":"official","path":"beds-bench/prepare.py","file_url":"https://github.com/google-health/records-research/blob/HEAD/beds-bench/prepare.py","link_basis":"first_harvest_node","language":"python","status":"ran_honours","verification_level":1,"contract_check":"HONOURS","metamorphic_tier":"well_formed","behaviour_fingerprint":true,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"cddb6efa0ce3d290"}},{"code_sha256_prefix":"a91f623210840678","entry":"config_get","repo":"google-health/records-research","repo_kind":"official","path":"beds-bench/prepare.py","file_url":"https://github.com/google-health/records-research/blob/HEAD/beds-bench/prepare.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"a91f623210840678"}}]},"arxiv_metadata":null,"syntology_extracted_results":null}