A self-evaluated predictive model: A Bayesian neural network approach to colorectal cancer diagnosis

Abstract

Artificial intelligence has shown immense potential in cancer prediction, but existing models cannot estimate prediction uncertainty by themselves. Here, we developed a Bayesian neural network (BNN) model, BNN-CRC₁₅, for colorectal cancer (CRC) prediction while assessing its reliability. The model was trained on routine laboratory data obtained from 27,911 participants and provided quantified prediction uncertainty, allowing identification of a subset of participants in which the model was confident, mimicking the diagnostic process of human practitioners. Our model exhibited superior performance (area under the curve = 0.918) in the confident participant group, which accounted for 46.4% of the patients, indicating that routine laboratory data alone are sufficient for accurate predictions in this subset. For the non-confident group, further advanced tests, such as colonoscopy, could be recommended to achieve more accurate predictions. In addition, our model demonstrated superior overall accuracy(0.848) in all patients, outperforming other five traditional algorithms (extreme gradient boosting, support vector machine, logistic regression, random forest, and artificial neural network) and fecal immunochemical test in distinguishing CRC from non-CRC. These findings suggest that our BNN-CRC₁₅ model could serve as a valuable tool for improving CRC diagnosis and prevention.

Keywords

artificial intelligence / Bayesian neural network / colorectal cancer / routine laboratory tests / uncertainty quantification

References

[1] M. E. Mccue, A. M. Mccoy, Front Vet. Sci. 2017, 4, 194.

[2] S. Shilo, H. Rossman, E. Segal, Nat. Med. 2020, 26, 29.

[3] B. H. Kann, A. Hosny, H. J. W. L. Aerts, Cancer Cell 2021, 39, 916.

[4] M. Saberi-Karimian, Z. Khorasanchi, H. Ghazizadeh, M. Tayefi, S. Saffar, G. A. Ferns, M. Ghayour-Mobarhan, Crit. Rev. Clin. Lab. Sci. 2021, 58, 275.

[5] D. Cirillo, A. Valencia, Curr. Opin. Biotechnol. 2019, 58, 161.

[6] C. Ma, X. Wang, J. Wu, X. Cheng, L. Xia, F. Xue, L. Qiu, Clin. Biochem. 2020, 84, 21.

[7] D. S. Herman, D. D. Rhoads, W. L. Schulz, T. J. S. Durant, Clin. Chem. 2021, 67, 1466.

[8] S. Abelson, G. Collord, S. W. K. Ng, O. Weissbrod, N. Mendelson Cohen, E. Niemeyer, N. Barda, P. C. Zuzarte, L. Heisler, Y. Sundaravadanam, R. Luben, S. Hayat, T. T. Wang, Z. Zhao, I. Cirlan, T. J. Pugh, D. Soave, K. Ng, C. Latimer, C. Hardy, L. I. Shlush, Nature 2018, 559, 400.

[9] B. Wang, J. Jing, X. Huang, C. Hua, Q. Qin, Y. Jia, Z. Wang, L. Jiang, B. Gao, L. Wu, X. Zeng, F. Wang, C. Mao, S. Liu,Adv. Intell. Syst. 2022, 4, 2100204.

[10] B. Wang, W. Wei, Z. Shao, Q. Qin, Z. Wang, Y. Jia, J. Guo, Y. Pang, L. Jiang, G. Jin, C. Mao, S. Liu, Adv. Intell. Syst. 2021, 3, 2100033.

[11] P. Doupe, J. Faghmous, S. Basu, Value Health 2019, 22, 808.

[12] J. G. Greener, S. M. Kandathil, L. Moffat, D. T. Jones, Nat. Rev. Mol. Cell Biol. 2022, 23, 40.

[13] S. Uddin, A. Khan, M. E. Hossain, M. A. Moni, BMC Med. Inform. Decis. Mak. 2019, 19, 281.

[14] M. M. Najafabadi, F. Villanustre, T. M. Khoshgoftaar, N. Seliya, R. Wald, E. A. Muharemagic, J. Big Data 2015, 2, 1.

[15] T. Jiang, J. L. Gradus, A. J. Rosellini, Behav. Ther. 2020, 51, 675.

[16] Y. Mae, W. Kumagai, T. Kanamori, Neural Netw. 2021, 144, 394.

[17] C. Bishop, Pattern Recognition and Machine Learning, Springer Press, New York 2006.

[18] H. Sung, J. Ferlay, R. L. Siegel, M. Laversanne, I. Soerjomataram, A. Jemal, F. Bray, CA Cancer J. Clin. 2021, 71, 209.

[19] P. Baglioni, N. Engl. J. Med. 2018, 379, 301.

[20] D. Heresbach, S. Manfredi, P. N. D’halluin, J. F. Bretagne, B. Branger, Eur. J. Gastroenterol. Hepatol. 2006, 18, 427.

[21] S. C. Toma, B. S. Ungureanu, S. Patrascu, V. Surlin, I. Georgescu, Curr. Health Sci. J. 2018, 44, 140.

[22] Y. Kinar, N. Kalkstein, P. Akiva, B. Levin, E. E. Half, I. Goldshtein, G. Chodick, V. Shalev, J. Am. Med. Inform. Assoc. 2016, 5, 879.

[23] F. Firat, A. K. Arslan, C. Colak, H. Harputluoglu, Kuwait J. Sci. 2016, 43, 151.

[24] B. Boursi, R. Mamtani, W. T. Hwang, K. Haynes, Y. X. Yang, Dig. Dis. Sci. 2016, 61, 2076.

[25] R. Goshen, B. Mizrahi, P. Akiva, Y. Kinar, E. Choman, V. Shalev, V. Sopik, R. Kariv, S. A. Narod, Br. J. Cancer 2017, 116, 944.

[26] G. S. Collins, J. B. Reitsma, D. G. Altman, K. G. Moons, BMJ 2015, 350, g7594.

[27] E. E. Low, J. Demb, L. Liu, A. Earles, R. Bustamante, C. D. Williams, D. Provenzale, T. Kaltenbach, A. J. Gawron, M. E. Martinez, S. Gupta, Gastroenterology 2020, 159, 492.

[28] A. N. Tarekegn, M. Giacobini, K. Michalak, Pattern Recognit. 2021, 118, 107965.

[29] Y. Kwon, J. H. Won, B. J. Kim, M. C. Paik, Comput. Stat. Data Anal. 2020, 142, 106816.

[30] S. Ryu, Y. Kwon, W. Y. Kim, Chem. Sci. 2019, 10, 8438.

[31] J. Gawlikowski, C. R. N. Tassi, M. Ali, J. Lee, X. X. Zhu, arXiv 2021, 2107, 03342.

[32] R. Grewal, J. A. Cote, H. Baumgartner, Marketing Sci. 2004, 4, 23.

[33] M. Hu, C. Asami, H. Iwakura, Y. Nakajima, R. Sema, T. Kikuchi, T. Miyata, K. Sakamaki, T. Kudo, M. Yamada, T. Akamizu, Y. Sakakibara, Commun. Med. 2022, 2, 9.

[34] B. J. Nartowt, G. R. Hart, D. A. Roffman, X. Llor, I. Ali, W. Muhammad, Y. Liang, J. Deng, PLoS One 2019, 14, e0221421.

[35] Y. Meng, J. Sun, J. Yu, C. Wang, J. Su,Biol. TraceElem. Res. 2019, 189, 325.

[36] D. You, M. Zhang, W. He, D. Wang, Y. Yu, Z. Yu, T. Lange, S. Yang, Y. Wei, H. Ma, Z. Hu, H. Shen, F. Chen, Y. Zhao, Transl. Lung Cancer Res. 2021, 10, 45.

[37] A. E. Prizment, R. A. Vierkant, T. C. Smyrk, L. S. Tillmans, J. J. Lee, P. Sriramarao, H. H. Nelson, C. F. Lynch, S. N. Thibodeau, T. R. Church, J. R. Cerhan, K. E. Anderson, P. J. Limburg, Mod. Pathol. 2016, 29, 516.

[38] A. Gupta, E. Gupta, R. Hilsden, J. D. Hawel, A. I. Elnahas, C. M. Schlachta, N. A. Alkhamesi, Can. J. Surg. 2021, 64, E621.

[39] Y. Li, H. Wu, C. Xing, X. Hu, F. Zhang, Y. Peng, Z. Li, T. Lu, J. Cancer 2020, 11, 3834.

[40] Y. Cao, S. Deng, L. Yan, J. Gu, J. Yang, M. Yang, L. Liu, K. Cai, BMC Cancer 2021, 21, 85.

[41] S. Ghuman, M. Van Hemelrijck, H. Garmo, L. Holmberg, H. Malmstrom, M. Lambe, N. Hammar, G. Walldius, I. Jungner, W. Wulaningsih, Br. J. Cancer 2017, 116, 1358.

[42] T. Yamamoto, K. Kawada, K. Obama, Int. J. Mol. Sci. 2021, 22, 8002.

RIGHTS & PERMISSIONS

2023 The Authors. View published by Shanghai Fuji Technology Consulting Co., Ltd, authorized by Professional Community of Experimental Medicine, National Association of Health Industry and Enterprise Management (PCEM) and John Wiley & Sons Australia, Ltd.