mlsanityData to 7 Oct 2026

A Nationwide Japanese Medical Claims Foundation Model: Balancing Model Scaling and Task-Specific Computational Efficiency

arXiv 2604.22348 ↗First posted 24 Apr 2026cs.LGcited by 0

Not on the map. Why?

Abstract (arXiv)

Clinical risk prediction using longitudinal medical data supports individualized care. Self-supervised foundation models have emerged as a promising approach for leveraging large-scale unlabeled healthcare records. In natural language processing, scaling laws suggest that larger models achieve predictably lower pretraining losses, supporting the foundation model paradigm. However, for structured medical data, characterized by a limited vocabulary and sparse observations, whether increasing model size consistently improves downstream predictions is unclear, as most studies evaluate only a single model scale. In this study, we evaluated the relationship between model scale and downstream task performance for structured medical foundation models. Using a random sample (2.3 million patients, 32 hospitals) from a nationwide 519-hospital Japanese claims database, we pretrained encoder-only Transformers at five scales (2.2M-101M parameters) for disease incidence and medication prediction. Downstream performance saturated at task-dependent thresholds: disease prediction benefited from larger models (32M-101M), whereas medication prediction saturated at 11M, reducing pretraining time by 178 h. Across all tasks, the best-performing model consistently outperformed a Light Gradient Boosting Machine baseline in the area under the precision-recall curve. These findings indicate that, unlike the monotonically decreasing pretraining loss, the optimal model size varied depending on task characteristics. This task-dependent saturation provides practical guidance for balancing predictive performance and computational cost in structured medical foundation models.

Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 18 of its 32 references that are in the map. First posted on arXiv 24 Apr 2026.

Builds on

18 of 32 references.

References · 18

  1. 2017Attention Is All You Needcited by 59,932
  2. 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,204
  3. 2020Scaling Laws for Neural Language Modelscited by 6,411
  4. 2021On the Opportunities and Risks of Foundation Modelscited by 3,961
  5. 2022Training Compute-Optimal Large Language Modelscited by 2,927
  6. 2022Large Language Models Encode Clinical Knowledgecited by 1,807
  7. 2021Revisiting Deep Learning Models for Tabular Datacited by 725
  8. 2020Med-BERT: pre-trained contextualized embeddings on large-scale structured electronic health records for disease predictioncited by 368
  9. 2018Scalable and accurate deep learning for electronic health recordscited by 275
  10. 2019BEHRT: Transformer for Electronic Health Recordscited by 258
  11. 2022GatorTron: A Large Clinical Language Model to Unlock Patient Information from Unstructured Electronic Health Recordscited by 238
  12. 2022On Embeddings for Numerical Features in Tabular Deep Learningcited by 185
6 more
  1. 2023EHRSHOT: An EHR Benchmark for Few-Shot Evaluation of Foundation Modelscited by 101
  2. 2021CEHR-BERT: Incorporating temporal information from structured EHR data to improve prediction taskscited by 75
  3. 2024Making Pre-trained Language Models Great on Tabular Predictioncited by 56
  4. 2022GenHPF: General Healthcare Predictive Framework with Multi-task Multi-source Learningcited by 35
  5. 2023A Multi-Center Study on the Adaptability of a Shared Foundation Model for Electronic Health Recordscited by 30
  6. 2025Generative Medical Event Models Improve with Scalecited by 24

Built on by

Not cited yet.