Optimizing risk-based breast cancer screening policies with reinforcement learning.
Adam YalaPeter G MikhaelConstance LehmanGigin LinFredrik StrandYung Liang WanKevin HughesSiddharth SatuluruThomas KimImon BanerjeeJudy GichoyaHari TrivediRegina BarzilayPublished in: Nature medicine (2022)
Screening programs must balance the benefit of early detection with the cost of overscreening. Here, we introduce a novel reinforcement learning-based framework for personalized screening, Tempo, and demonstrate its efficacy in the context of breast cancer. We trained our risk-based screening policies on a large screening mammography dataset from Massachusetts General Hospital (MGH; USA) and validated this dataset in held-out patients from MGH and external datasets from Emory University (Emory; USA), Karolinska Institute (Karolinska; Sweden) and Chang Gung Memorial Hospital (CGMH; Taiwan). Across all test sets, we find that the Tempo policy combined with an image-based artificial intelligence (AI) risk model is significantly more efficient than current regimens used in clinical practice in terms of simulated early detection per screen frequency. Moreover, we show that the same Tempo policy can be easily adapted to a wide range of possible screening preferences, allowing clinicians to select their desired trade-off between early detection and screening costs without training new policies. Finally, we demonstrate that Tempo policies based on AI-based risk models outperform Tempo policies based on less accurate clinical risk models. Altogether, our results show that pairing AI-based risk models with agile AI-designed screening policies has the potential to improve screening programs by advancing early detection while reducing overscreening.
Keyphrases
- public health
- artificial intelligence
- healthcare
- end stage renal disease
- machine learning
- mental health
- clinical practice
- chronic kidney disease
- big data
- high resolution
- computed tomography
- peritoneal dialysis
- human health
- climate change
- newly diagnosed
- emergency department
- high throughput
- body composition
- decision making
- rna seq
- breast cancer risk
- image quality
- resistance training