Back

Synthetic Histology Images for Training AI Models: A Novel Approach to Improve Prostate Cancer Diagnosis

Van Booven, D. J.; Chen, C.-B.; Kryvenko, O.; Punnen, S.; Sandoval, V.; Malpani, S.; Noman, A.; Ismael, F.; Briseno, A.; Wang, Y.; Arora, H.

2024-01-29 cancer biology
10.1101/2024.01.25.577225 bioRxiv
Show abstract

Prostate cancer (PCa) poses significant challenges for timely diagnosis and prognosis, leading to high mortality rates and increased disease risk and treatment costs. Recent advancements in machine learning and digital imagery offer promising potential for developing automated and objective assessment pipelines that can reduce human capital and resource costs. However, the reliance of AI models on large amounts of clinical data for training presents a significant challenge, as this data is often biased, lacking diversity, and not readily available. Here we aim to address this limitation by employing customized generative adversarial network (GAN) models to produce high-quality synthetic images of different PCa grades (radical prostatectomy (RP)) and needle biopsies, which were customized to account for the granularity associated with each Gleason grade. The generated images were subjected to multiple rounds of benchmarking, quantifications and quality control assessment before being used to train an AI model (EfficientNet) for grading digital histology images of adenocarcinoma specimens (RP sections) and needle biopsies obtained from the PANDA challenge repository. Validation was performed using the AI model trained with synthetic data to grade digital histology from the cancer genome atlas (TCGA) (RP sections) and needle biopsy data from Radboud University Medical Center and Karolinska Institute. Results demonstrated that the AI model trained with a combination of image patches derived from original and enhanced synthetic images outperformed the model trained with original digital histology images. Together, this study demonstrates the potential of customized GAN models to generate a large cohort of synthetic data that can train AI models to effectively grade PCa specimens. This approach could potentially eliminate the need for extensive clinical data for training any AI model in the domain of digital imagery, leading to cost and time-effective diagnosis and prognosis.

Matching journals

The top 8 journals account for 50% of the predicted probability mass.

1
Journal of Medical Imaging
11 papers in training set
Top 0.1%
12.5%
2
Scientific Reports
3612 papers in training set
Top 5%
9.7%
3
PLOS ONE
5266 papers in training set
Top 25%
6.6%
4
PLOS Computational Biology
1863 papers in training set
Top 6%
6.2%
5
Journal of Translational Medicine
57 papers in training set
Top 0.1%
4.8%
6
Computers in Biology and Medicine
128 papers in training set
Top 0.7%
4.8%
7
Medical Physics
14 papers in training set
Top 0.1%
4.3%
8
Modern Pathology
22 papers in training set
Top 0.1%
4.0%
50% of probability mass above
9
npj Precision Oncology
53 papers in training set
Top 0.4%
3.4%
10
Cancers
213 papers in training set
Top 2%
3.2%
11
JCO Clinical Cancer Informatics
22 papers in training set
Top 0.3%
3.1%
12
npj Digital Medicine
118 papers in training set
Top 2%
2.4%
13
Computational and Structural Biotechnology Journal
242 papers in training set
Top 2%
2.4%
14
Journal of Pathology Informatics
15 papers in training set
Top 0.1%
1.9%
15
Biology Methods and Protocols
61 papers in training set
Top 0.7%
1.9%
16
Frontiers in Oncology
103 papers in training set
Top 2%
1.9%
17
BMC Cancer
67 papers in training set
Top 1%
1.7%
18
Diagnostics
50 papers in training set
Top 1%
1.7%
19
Patterns
78 papers in training set
Top 2%
1.3%
20
Frontiers in Bioinformatics
49 papers in training set
Top 0.9%
1.1%
21
Cancer Research Communications
51 papers in training set
Top 2%
1.0%
22
Breast Cancer Research
36 papers in training set
Top 0.5%
1.0%
23
npj Systems Biology and Applications
125 papers in training set
Top 2%
1.0%
24
GigaScience
212 papers in training set
Top 4%
1.0%
25
Bioinformatics
1204 papers in training set
Top 8%
1.0%
26
Communications Medicine
113 papers in training set
Top 5%
0.8%
27
Computer Methods and Programs in Biomedicine
28 papers in training set
Top 1%
0.8%
28
Communications Biology
993 papers in training set
Top 31%
0.8%
29
Nature Communications
5641 papers in training set
Top 57%
0.8%
30
Cells
249 papers in training set
Top 9%
0.6%