TSR-Driven CNN Optimization for Accurate and Interpretable Nonalcoholic Fatty Liver Disease Diagnosis
Abstract
Nonalcoholic Fatty Liver Disease (NAFLD) has emerged as one of the most prevalent chronic liver disorders worldwide, closely associated with sedentary lifestyles, obesity, and metabolic dysfunctions. Early detection is challenging due to the asymptomatic nature of initial stages and variability in imaging quality. Conventional ultrasound-based diagnosis is limited by operator dependency and subjective interpretation, while manual feature extraction and classical machine learning approaches often fail to capture subtle hepatic textural variations, restricting sensitivity in early-stage disease. This study proposes a fully automated, hybrid framework for NAFLD assessment from ultrasound images, integrating Convolutional Neural Networks (CNNs) with Tree-Structured Regularization (TSR) and metaheuristic optimization. CNNs enable hierarchical, data-driven feature extraction, while TSR imposes a biologically inspired hierarchical structure on features, enhancing interpretability and preventing overfitting. Metaheuristic optimization algorithms further fine-tune hyperparameters and select optimal feature subsets, improving both accuracy and model generalization. The framework emphasizes robustness across heterogeneous ultrasound systems, high sensitivity in mild steatosis, and computational efficiency suitable for real-time applications. Experimental evaluations demonstrate that TSR-optimized CNNs outperform traditional optimization methods, achieving higher classification accuracy, faster convergence, and increased resilience to noise. Feature activation analyses indicate improved discriminative representation, confirming the effectiveness of hierarchical optimization in guiding CNN learning. The hybrid framework reduces reliance on invasive diagnostic procedures and supports objective, reproducible, and clinically meaningful assessment of hepatic steatosis.
References
- [1] Deng, L., Huang, J., Yuan, H., Liu, Q., Lou, W., Yu, P., … & Deng, L. (2025). Biological age prediction and NAFLD risk assessment: a machine learning model based on a multicenter population in Nanchang, Jiangxi, China. BMC gastroenterology, 25(1), 1-12. https://doi.org/10.1186/s12876-025-03752-y
- [2] Wang, Y., & Wang, P. (2025). Development and validation of a new diagnostic prediction model for NAFLD based on machine learning algorithms in NHANES 2017-2020.3. Hormones, 24(2), 461–476. https://doi.org/10.1007/s42000-025-00634-6
- [3] Tian, F., Zhao, Y., He, X., Zhang, Y., Hu, M., Liang, Y., … & Wang, Y. (2025). Interpretable Machine learning reveals the role of PANoptosis in the diagnosis and subtyping of NAFLD. Immunobiology, 230(3), 1-18. https://doi.org/10.1016/j.imbio.2025.152909
- [4] Charu, V., Liang, J. W., Mannalithara, A., Kwong, A., Tian, L., & Kim, W. R. (2024). Benchmarking clinical risk prediction algorithms with ensemble machine learning for the noninvasive diagnosis of liver fibrosis in NAFLD. Hepatology, 80(5), 1184–1195. https://doi.org/10.1097/HEP.0000000000000908
- [5] Thatha, V. N., Chalichalamala, S., Pamula, U., Krishna, D. P., Chinthakunta, M., Mantena, S. V., … & Vatambeti, R. (2025). Optimized machine learning mechanism for big data healthcare system to predict disease risk factor. Scientific reports, 15(1), 14327. https://doi.org/10.1038/s41598-025-98721-6
- [6] Aliyu, D. A., Akhir, E. A. P., Osman, N. A., Salisu, J. A., Saidu, Y., & Yalli, J. S. (2024). Optimization techniques in reinforcement learning for healthcare: a review. 2024 8th international conference on computing, communication, control and automation (ICCUBEA) (pp. 1–6). IEEE. https://doi.org/10.1109/ICCUBEA61740.2024.10774698
- [7] Goswami, M., Mohanty, S., & Pattnaik, P. K. (2024). Optimization of machine learning models through quantization and data bit reduction in healthcare datasets. Franklin open, 8, 1-11. https://doi.org/10.1016/j.fraope.2024.100136
- [8] Ma, D., Zheng, W., & Lu, T. (2024). Machine learning-based predictive model for service quality assessment and policy optimization in adult day health care centers. Spectrum of research, 4(1), 1–14. https://doi.org/10.1002/aic.16729
- [9] Fatima, S. (2024). Healthcare cost optimization: leveraging machine learning to identify inefficiencies in healthcare systems. International journal of advanced research in engineering technology & science, 10(3), 137–147. https://B2n.ir/ry4653
- [10] Olalekan Kehinde, A. (2025). Leveraging machine learning for predictive models in healthcare to enhance patient outcome management. International research journal of modernization in engineering technology and science, 7(1), 1465–1482. https://www.doi.org/10.56726/IRJMETS66198
- [11] Khaleel, I., Marzoog, W. N., & Al-Kateb, G. (2025). ANILA: adaptive neuro-inspired learning algorithm for efficient machine learning, AI optimization, and healthcare enhancement. Mesopotamian journal of computer science, 2025, 159–171. https://doi.org/10.58496/MJCSC/2025/009
- [12] Krishnan, P. (2024). Ai-driven optimization in healthcare: machine learning models for predictive diagnostics and personalized treatment strategies. Well testing journal, 33(S2), 10–33. https://welltestingjournal.com/index.php/WT/article/view/Ai_Driven_Optimization_In_Healthcar_Machine_Learning_Models_For_
- [13] Nadeem, M. I., Ahmed, K., Li, D., Zheng, Z., Naheed, H., Muaad, A. Y., … & Abdel Hameed, H. (2022). SHO-CNN: a metaheuristic optimization of a convolutional neural network for multi-label news classification. Electronics, 12(1), 1-24. https://doi.org/10.3390/electronics12010113
- [14] Daliri, A., Branch, K., Sheikha, M., Roudposhti, K. K., Branch, L., Alimoradi, M., & Mohammadzadeh, J. (2023). Optimized categorical boosting for gastric cancer classification using heptagonal reinforcement learning and the water optimization algorithm. 7th international conference on pattern recognition and image analysis (IPRIA) (pp. 1–6). IEEE. https://www.researchgate.net/profile/Arman-Daliri/publication/389465487_Optimized_Categorical_Boosting_for_Gastric_Cancer_Classification_using_Heptagonal_Reinforcement_learning_and_the_Water_Optimization_Algorithm/links/67c3173f645ef274a4988598/Optimized-
- [15] Taji, K., Sohail, A., Shahzad, T., Khan, B. S., Khan, M. A., & Ouahada, K. (2024). An ensemble hybrid framework: a comparative analysis of metaheuristic algorithms for ensemble hybrid CNN features for plants disease classification. IEEE access, 12, 61886–61906. https://doi.org/10.1109/ACCESS.2024.3389648
- [16] Alimoradi, M. (2018). Investigating the composition of apriori algorithm and metaheuristic algorithms (Genetic and PSO). https://doi.org/10.13140/RG.2.2.14058.77761
- [17] He, K., Zhang, X., Ren, S., & Sun, J. (2016). Deep residual learning for image recognition. Proceedings of the ieee conference on computer vision and pattern recognition (pp. 770–778). Computer Vision and Pattern Recognition. https://doi.org/10.1109/cvpr.2016.90
- [18] Nagoor, S., & Jinny, S. V. (2023). A dual fuzzy with hybrid deep learning architecture based on CNN with hybrid metaheuristic algorithm for effective segmentation and classification. International journal of information technology, 15(1), 531–543. https://doi.org/10.1007/s41870-022-01106-5
- [19] Prakash, N. N., Rajesh, V., Namakhwa, D. L., Dwarkanath Pande, S., & Ahammad, S. H. (2023). A DenseNet CNN-based liver lesion prediction and classification for future medical diagnosis. Scientific african, 20, 1-16. https://doi.org/10.1016/j.sciaf.2023.e01629
- [20] Rikhtegar, A., Pooyan, M., & Manzuri, M. T. (2020). Comparing performance of metaheuristic algorithms for finding the optimum structure of CNN for face recognition. International journal of nonlinear analysis and applications, 11(1), 301–319. https://doi.org/10.22075/ijnaa.2020.4296
- [21] Naderi Yaghouti, A. R., & Shalbaf, A. (2024). Classification of nonalcoholic fatty liver grades using pre-trained convolutional neural networks and a random forest classifier on B-mode ultrasound images. Journal of biomedical physics and engineering, 15(5), 437–450. https://doi.org/10.31661/jbpe.v0i0.2307-1646
- [22] Alimoradi, M., Zabihimayvan, M., Daliri, A., Sledzik, R., & Sadeghi, R. (2022). Deep neural classification of darknet traffic. In Artificial intelligence research and development (pp. 105–114). IOS press. https://doi.org/10.3233/FAIA220323
- [23] Daliri, A., Asghari, A., Azgomi, H., & Alimoradi, M. (2022). The water optimization algorithm: A novel metaheuristic for solving optimization problems. Applied intelligence, 52(15), 17990–18029. https://doi.org/10.1007/s10489-022-03397-4
- [24] Fakheri, S., Alimoradi, M., & Yamaghani, M. R. (2024). Colour image multilevel thresholding segmentation using trees social relationship algorithm. Research square, Preprint. https://doi.org/10.21203/rs.3.rs-4479475/v1
- [25] Alimoradi, M., Sadeghi, R., Daliri, A., & Zabihimayvan, M. (2025). Statistic deviation mode balancer (SDMB): A novel sampling algorithm for imbalanced data. Neurocomputing, 624, 129484. https://doi.org/10.1016/j.neucom.2025.129484
- [26] Alimoradi, M., Azgomi, H., & Asghari, A. (2022). Trees social relations optimization algorithm: A new swarm-based metaheuristic technique to solve continuous and discrete optimization problems. Mathematics and computers in simulation, 194, 629–664. https://doi.org/10.1016/j.matcom.2021.12.010