Adaptive neuro-fuzzy inference system classifier with interpretability for cancer diagnostic50 views
Keywords:Neuro-fuzzy network; Attention; iANFIS; Interpretable AI; Cancer diagnostic.
Clinical outcome analysis using patient medical data facilitates clinical decision-making and increases prognostic accuracy. Recently, deep learning (DL) with learning big data features has shown expert-level accuracy in predicting clinical outcomes. Many of these sophisticated machine learning models, however, lack interpretability, creating significant trust-related healthcare issues. This necessarily requires the need for interpretable AI systems capable of explaining their decisions. In this respect, the paper proposes an interpretable classifier of the adaptive neuro-fuzzy inference method (iANFIS), which combines the fuzzy inference system with critical rule selection by attention mechanism. The rule-based processing of ANFIS helps the user to understand the behavior of the proposed model. The essential activated rule and the most important input features that predict the outcome are identified by the attention-based rule selector. We conduct two experiments with two cancer diagnostic datasets for verifying the performance of the proposed iANFIS. By using recursive rule elimination (RRE) to prune fuzzy rules, the model’s complexity is significantly reduced while preserving system efficiency that makes it more interpretable.
. K. Kourou, T. P. Exarchos, K. P. Exarchos, M. V. Karamouzis, and D. I. Fotiadis, “Machine learning applications in cancer prognosis and prediction,” Computational and structural biotechnology journal, vol. 13, pp. 8–17, (2015). DOI: https://doi.org/10.1016/j.csbj.2014.11.005
. P. Danaee, R. Ghaeini, and D. A. Hendrix, “A deep learning approach for cancer detection and relevant gene indentification.” Pac. Symp. Biocomput., vol. 22, pp. 219–229, (2017).
. J. Grundner, H. U. Prokosch, M. St??rzl, R. Croner, J. Christoph, and D. Toddenroth, “Predicting clinical outcomes in colorectal cancer using machine learning.” Stud Health Technol Inform, vol. 247, pp. 101–105, (2018).
. R. Al-Bahrani, A. Agrawal, and A. Choudhary, “Survivability prediction of colon cancer patients using neural networks,” Health Informatics J., (2017). DOI: https://doi.org/10.1177/1460458217720395
. K. M. Yang, C. S. Yu, J. L. Lee, C. W. Kim, Y. S. Yoon, I. J. Park, S.-B. Lim, S. H. Park, B. D. Ye, S.-K. Yang, and J. C. Kim, “Risk factors for postoperative recurrence after primary bowel resection in patients with crohn??s disease,” World J Gastroenterol., vol. 23, no. 38, p. 7016??7024, (2017). DOI: https://doi.org/10.3748/wjg.v23.i38.7016
. S. Guillaume, “Designing fuzzy inference systems from data: An interpretability-oriented review,” IEEE Transactions on fuzzy systems, vol. 9, no. 3, pp. 426–443, (2001). DOI: https://doi.org/10.1109/91.928739
. J.-S. Jang, “Anfis: adaptive-network-based fuzzy inference system” IEEE transactions on systems, man, and cybernetics, vol. 23, pp. 665–685, (1993). DOI: https://doi.org/10.1109/21.256541
. T. Takagi and M. Sugeno, “Fuzzy Identification of Systems and Its Applications to Modeling and Control,” IEEE Transactions on Systems, Man, and Cybernetics, vol. SMC-15, no. 1, pp. 116–132, (1985). DOI: https://doi.org/10.1109/TSMC.1985.6313399
. M.-T. Luong, H. Pham, and C. D. Manning, “Effective approaches to attention-based neural machine translation,” arXiv preprint arXiv:1508.04025, (2015). DOI: https://doi.org/10.18653/v1/D15-1166
. D. P. Kingma and J. Ba, “Adam: A method for stochastic optimization,” arXiv preprint arXiv:1412.6980, (2014).
. T. Akiba, S. Sano, T. Yanase, T. Ohta, and M. Koyama, “Optuna: A next-generation hyperparameter optimization framework,” in Proceedings of the 25th ACM SIGKDD International Conference on Knowledge Discovery & Data Mining, pp. 2623–2631, (2019). DOI: https://doi.org/10.1145/3292500.3330701
. G. E. Batista, R. C. Prati, and M. C. Monard, “A study of the behavior of several methods for balancing machine learning training data,” ACM SIGKDD explorations newsletter, vol. 6, no. 1, pp. 20–29, (2004). DOI: https://doi.org/10.1145/1007730.1007735
. K. Bache and M. Lichman, “Uci machine learning repository,” (2013).