Hybrid Graph Mamba Framework for Histopathological Image Classification
PDF
PDF

How to Cite

Machnaim, Dave, Shawn Jaison, Naveen Sundar, Narmadha Naveen D., and Sai Midhun. 2026. “Hybrid Graph Mamba Framework for Histopathological Image Classification”. Journal of Trends in Computer Science and Smart Technology 8 (2): 304-23. https://doi.org/10.36548/jtcsst.2026.2.006.

Keywords

Computational Pathology
Histopathological Image Analysis
Selective State-Space Models (Mamba)
Liquid Neural ODEs
Graph Neural Networks
Spectral Representation Learning
Functional Kernel Adaptive Networks (KAN)
Explainable-AI(XAI)

Abstract

Digital pathology has already been shown to be useful in disease diagnosis; however, the computational power required to use high-resolution digital pathology is substantial, making it unrealistic in many clinical laboratories due to limitations in hardware capabilities. Accordingly, this paper presents a new architecture, MFLA-Graph-Mamba-NCA, which addresses these limitations. The main point of MFLA-Graph-Mamba-NCA is to develop a lightweight model for tissue classification while maintaining speed and optimal accuracy. Four complementary learning paradigms are incorporated into the architecture:  a morphological frontend, a frequency-domain branch based on Functional Kernel Adaptive Networks (FunKAN), feature stabilization over time using Liquid Neural Ordinary Differential Equations, and global sequence modelling using a Graph-based Mamba module. The model was tested on a harmonized benchmark of 125,000 images, based on the NCT-CRC-HE-100K and LC25000 datasets, and divided into 12 different biological classes to expressly avoid cross-dataset lab leakage. It has been shown to provide 99.89%accuracy and a Macro: F1 score of 99.78%. These techniques were condensed to 1.46 million parameters, forming a hybridized model that can be deployed to standard clinical workstations.

References

  1. Ke, Qi, Wun-She Yap, Yee Kai Tee, Yan Chai Hum, Hua Zheng, and Yu-Jian Gan. "Advanced Deep Learning for Multi-Class Colorectal Cancer Histopathology: Integrating Transfer Learning and Ensemble Methods." Quantitative imaging in medicine and surgery 15, no. 3 (2025): 2329-2346.
  2. AlGhamdi, Rayed, Turky Omar Asar, Fatmah Y. Assiri, Rasha A. Mansouri, and Mahmoud Ragab. "Al-Biruni Earth Radius Optimization with Transfer Learning Based Histopathological Image Analysis for Lung and Colon Cancer Detection." Cancers 15, no. 13 (2023): 3300.
  3. MD, Aaseegha, and Venkataramana B. "A Hybrid Framework for Enhanced Segmentation and Classification of Colorectal Cancer Histopathology." Frontiers in Artificial Intelligence 8 (2025): 1647074.
  4. Rewatkar, Rasika M., AR Bhagat Patil, and Ashutosh Bagde. "A Comprehensive Evaluation of Data Analysis Approaches for Predicting Colorectal Cancer." In AIP Conference Proceedings, vol. 3188, no. 1, AIP Publishing LLC, 2024, 100094.
  5. ALGhafri, Houda Saif, and Chia S. Lim. "Fine-Tuning Models for Histopathological Classification of Colorectal Cancer." Diagnostics 15, no. 15 (2025): 1947.
  6. Ke, Qi, Yan Chai Hum, Wun-She Yap, Tian Swee Tan, Humaira Nisar, Hamam Mokayed, AiQuan Li, Rong Gao, and YuJian Gan. "Histopathological Classification of Colorectal Cancer Based on Domain-Specific Transfer Learning and Multi-Model Feature Fusion." Scientific Reports 15, no. 1 (2025): 35155.
  7. Prakash, U. M., S. Iniyan, Ashit Kumar Dutta, Shtwai Alsubai, Janjhyam Venkata Naga Ramesh, Sachi Nandan Mohanty, and Khasim Vali Dudekula. "Multi-Scale Feature Fusion of Deep Convolutional Neural Networks on Cancerous Tumor Detection and Classification Using Biomedical Images." Scientific Reports 15, no. 1 (2025): 1105.
  8. Sharkas, Maha, and Omneya Attallah. "Color-CADx: A Deep Learning Approach for Colorectal Cancer Classification Through Triple Convolutional Neural Networks and Discrete Cosine Transform." Scientific Reports 14, no. 1 (2024): 6914.
  9. AlShehri, Helala. "Dual-Path Convolutional Neural Network with Squeeze-and-Excitation Attention for Lung and Colon Histopathology Classification." Journal of Imaging 11, no. 12 (2025): 448.
  10. Venkatraman, Shravan, Jaskaran Singh Walia, and P. R. Joe Dhanith. "SAG-ViT: A Scale-Aware, High-Fidelity Patching Approach with Graph Attention for Vision Transformers." Complex & Intelligent Systems 11, no. 10 (2025): 428.
  11. Li, Meng. "Transformer-Based Self-Supervised Learning and Distillation for Medical Image Classification: Improving Colorectal Cancer Detection on nct-crc-he-100k With Swin-T V2." In 2024 3rd International Conference on Cloud Computing, Big Data Application and Software Engineering (CBASE), IEEE, 2024, 644-648.
  12. Touvron, Hugo, Matthieu Cord, Matthijs Douze, Francisco Massa, Alexandre Sablayrolles, and Hervé Jégou. "Training Data-Efficient Image Transformers & Distillation Through Attention." In International conference on machine learning, PMLR, 2021, 10347-10357.
  13. Debnath, Jesika, Amira Hossain, Anamul Sakib, Hamdadur Rahman, Rezaul Haque, Md Redwan Ahmed, Ahmed Wasif Reza, SM Masfequier Rahman Swapno, and Abhishek Appaji. "LMVT: A Hybrid Vision Transformer with Attention Mechanisms for Efficient and Explainable Lung Cancer Diagnosis." Informatics in Medicine Unlocked (2025): 101669.
  14. Liu, Qifeng, Tao Zhou, Chi Cheng, Jin Ma, and Marzia Hoque Tania. "Hybrid Generative Adversarial Network Based on Frequency and Spatial Domain for Histopathological Image Synthesis." BMC bioinformatics 26, no. 1 (2025): 29.
  15. Karn, Prakash Kumar, Iman Ardekani, and Waleed H. Abdulla. "Generalized Framework for Liquid Neural Network Upon Sequential and Non-Sequential Tasks." Mathematics 12, no. 16 (2024): 2525.
  16. Gu, Albert, and Tri Dao. "Mamba: Linear-Time Sequence Modeling with Selective State Spaces." arXiv preprint arXiv:2312.00752 (2023).
  17. Liao, Jianshang, and Liguo Wang. "HyperspectralMamba: A Novel State Space Model Architecture for Hyperspectral Image Classification." Remote Sensing 17, no. 15 (2025): 2577.
  18. Sun, Yongheng, Yueh Z. Lee, Genevieve A. Woodard, Hongtu Zhu, Chunfeng Lian, and Mingxia Liu. "R2gen-mamba: A Selective State Space Model for Radiology Report Generation." In 2025 IEEE 22nd International Symposium on Biomedical Imaging (ISBI), IEEE, 2025, 1-4.
  19. Brussee, Siemen, Giorgio Buzzanca, Anne MR Schrader, and Jesper Kers. "Graph Neural Networks in Histopathology: Emerging Trends and Future Directions." Medical Image Analysis 101 (2025): 103444.
  20. Liu, Ziming, Yixuan Wang, Sachin Vaidya, Fabian Ruehle, James Halverson, Marin Soljačić, Thomas Y. Hou, and Max Tegmark. "Kan: Kolmogorov-Arnold Networks." arXiv preprint arXiv:2404.19756 (2024).
  21. Hasani, Ramin, Mathias Lechner, Alexander Amini, Daniela Rus, and Radu Grosu. "Liquid Time-Constant Networks." In Proceedings of the AAAI conference on artificial intelligence, vol. 35, no. 9, 2021, 7657-7666.
  22. Islam, Naeem Ul, Chung Tien Dat, and Muhammad Khalid. "Kolmogorov-Arnold Inspired Convolutional Networks for Enhancing PPO-based Online Reinforcement Learning." Journal of Experimental & Theoretical Artificial Intelligence 38, no. 2 (2026): 155-171.
  23. Opee, Shaiful Ajam, Arifa Akter Eva, Ahmed Taj Noor, Sayem Mustak Hasan, and M. F. Mridha. "ELW‐CNN: An Extremely Lightweight Convolutional Neural Network for Enhancing Interoperability in Colon and Lung Cancer Identification Using Explainable AI." Healthcare Technology Letters 12, no. 1 (2025): e12122.
  24. Riechie, Riechie, Vira Jessica, Matthew Kurniawan, and Feliks Victor Parningotan Samosir. "Convolutional Kolmogorov-Arnold Network for Pneumonia Detection in Medical Image Analysis." Indonesian Journal of Electronics, Electromedical Engineering, and Medical Informatics 7, no. 3 (2025): 475-487.
  25. Ignatov, Andrey, and Grigory Malivenko. "Nct-crc-he: Not All Histopathological Datasets are Equally Useful." In European Conference on Computer Vision, Cham: Springer Nature Switzerland, 2024, 300-317.
  26. Maheshwari, Ullagadi, B. V. Kiranmayee, and Chalumuru Suresh. "Diagnose Colon and Lung Cancer Histopathological Images Using Pre-Trained Machine Learning Model." In 2022 5th International Conference on Contemporary Computing and Informatics (IC3I), IEEE, 2022, 1078-1082.
  27. Sharma, Poonam, Bhisham Sharma, Ajit Noonia, Dhirendra Prasad Yadav, and Panos Liatsis. "Diagnosis of Colorectal Cancer Using Residual Transformer with Mixed Attention and Explainable AI." PLoS One 20, no. 11 (2025): e0335418.
  28. Amponsah, Anthony Afriyie. "Explainable AI for Histopathological Diagnosis: Comparison of Attention-Based Deep Learning Models with Expert Pathologists in Rare Tumor Classification." Journal of Integrated Research (2025): 102-131.
  29. Mothkur, Rashmi, and Pullagura Soubhagyalakshmi. "Grad-CAM Based Visualization for Interpretable Lung Cancer Categorization Using Deep CNN Models." Journal of Electronics, Electromedical Engineering, and Medical Informatics 7, no. 3 (2025): 567-580.
  30. Yan, Hongjie, Vivens Mubonanyikuzo, Temitope Emmanuel Komolafe, Liang Zhou, Tao Wu, and Nizhuan Wang. "Hybrid-RViT: Hybridizing ResNet-50 and Vision Transformer for Enhanced Alzheimer’s Disease Detection." PloS one 20, no. 2 (2025): e0318998.
  31. Dosovitskiy, Alexey, Lucas Beyer, Alexander Kolesnikov, Dirk Weissenborn, Xiaohua Zhai, Thomas Unterthiner, Mostafa Dehghani et al. "An Image is Worth 16x16 Words: Transformers for Image Recognition at Scale." arXiv preprint arXiv:2010.11929 (2020).
  32. Ochoa-Ornelas, Raquel, Alberto Gudiño-Ochoa, Sergio Octavio Rosales-Aguayo, Jesús Ezequiel Molinar-Solís, Sonia Espinoza-Morales, and René Gudiño-Venegas. "A Lightweight Cross-Gated Dual-Branch Attention Network for Colon and Lung Cancer Diagnosis from Histopathological Images." Medical Sciences 13, no. 4 (2025): 286.
  33. Said, Menatalla MR, Md Sakib Bin Islam, Md Shaheenur Islam Sumon, Semir Vranic, Rafif Mahmood Al Saady, Abdulrahman Alqahtani, Muhammad EH Chowdhury, and Shona Pedersen. "Innovative Deep Learning Architecture for the Classification of Lung and Colon Cancer from Histopathology Images." Applied Computational Intelligence and Soft Computing 2024, no. 1 (2024): 5562890.
  34. Kather, Jakob Nikolas, Niels Halama, and Alexander Marx. "100,000 Histological Images of Human Colorectal Cancer and Healthy Tissue." " https://www.kaggle.com/datasets/imrankhan77/nct-crc-he-100k (2018).
  35. Borkowski, Andrew A., Marilyn M. Bui, L. Brannon Thomas, Catherine P. Wilson, Lauren A. DeLand, and Stephen M. Mastorides. "Lung and Colon Cancer Histopathological Image Dataset (lc25000)." arXiv preprint arXiv:1912.12142 (2019).