Computer Science ›› 2026, Vol. 53 ›› Issue (7): 24-33.doi: 10.11896/jsjkx.250700003

• Computer Graphics & Multimedia • Previous Articles     Next Articles

AETC:Image Classification Model via Attention-based Topological Features Fusion

ZHU Bin, LI Xiaobin   

  1. School of Mathematics,Southwest Jiaotong University,Chengdu 611756,China
  • Received:2025-07-01 Revised:2025-08-29 Online:2026-07-15 Published:2026-07-10
  • About author:ZHU Bin,born in 2000,postgraduate,is a member of CCF(No.Z4495G).His main research interests include topolo-gical data analysis and image proces-sing.
    LI Xiaobin,born in 1983,Ph.D,mas-ter's supervisor,is a member of CCF(No.Z1644M).His main research interest is geometric topology and its applications.
  • Supported by:
    Fundamental Research Funds for the Central Universities of Ministry of Education of China(2682025ZTPY001,2682021ZTPY043) and National Natural Science Foundation of China(11501470,11426187).

Abstract: A novel attention-enhanced topology and convolution model,which combines persistent homology and convolutional neural network through an attention-guided fusion framework,is proposed to address degradation in classification performance caused by insufficient topological feature extraction and weak intra-class structural consistency.Persistent homology captures key topological structures and encodes them into feature descriptors,while convolutional neural network extracts local visual features through convolution and pooling operations.An attention mechanism then merges both into a unified global representation to enhance feature completeness.The Wasserstein distance cross entropy loss function,derived by integrating the measurable Wasserstein distance with the cross-entropy loss,is used to constrain the topological structures of images.This effectively mitigates inter-class topological ambiguity,thereby enhancing the classification performance,robustness,and accuracy of the AETC model.Models with various topological vectorization methods are evaluated on three diverse datasets,the AETC model improves ACC by 2%~11%,AUC by 1%~7%,F1-score by 1%~11%,and mAP by 3%~17%.Within the classical convolutional neural network framework enhanced by persistence landscape vectorization,the optimal model achieves peak ACC of 95.49%,AUC of 99.44%,F1-score of 95.48%,and mAP of 98.42%.

Key words: Persistent homology, Topological features vectorization, Image feature fusion, Attention mechanism, Convolutional neural network, Image classification

CLC Number: 

  • TP751.1
[1]LIU H D,SUN X H,LI Y B,et al.Review of Deep Learning Models for Image Classification Based on Convolutional Neural Networks [J].Chinese Journal of Computer Engineering and Applications,2025,61(11):1-21.
[2]GU J X,WANG Z H,KUEN J,et al.Recent Advances in Convolutional Neural Networks [J].Pattern Recognition,2018,77:354-377.
[3]LI D,WU H J,WANG Y J,et al.Lightweight Parallel Octave Convolutional Neural Network for Hyperspectral Image Classification[J].Photogrammetric Engineering & Remote Sensing,2023,89(4):233-243.
[4]KONG L J,WANG X W,BAO Y C,et al.ASurvey on Medical Image Segmentation Based on Deep Learning[J].Chinese Journal of Radio Communications Technology,2021,47(2):121-130.
[5]SUN Y N,XUE B,ZHANG M J,et al.Evolving Deep Convolutional Neural Networks for Image Classification[J].IEEE Transactions on Evolutionary Computation,2020,24(2):394-407.
[6]YAN J W,LIN X K,PAN Y Z.Fire image recognition method based on sparse convolutional network pruning [J].Journal of Chongqing Technology and Business University(Natural Science Edition) 2025,42(1):79-84.
[7]LECUN Y,BOTTOU L,BENGIO Y,et al.Gradient-based Learning Applied to Document Recognition[J].Proceedings of the IEEE,1998,86(11):2278-2324.
[8]SIMONYAN K,ZISSERMAN A.Very Deep Convolutional Networks for Large-scale Image Recognition[C]//Proceedings of the 3rd International Conference on Learning Representations.2015.
[9]HE K M,ZHANG X Y,REN S Q,et al.Deep Residual Learning for Image Recognition[C]//Proceedings of the 2016 IEEE Conference on Computer Vision and Pattern Recognition.Pisca-taway,NJ:IEEE,2016:770-778.
[10]HUANG G,LIU Z,VAN DER MAATEN L,et al.Densely Connected Convolutional Networks[C]//Proceedings of the 2017 IEEE Conference on Computer Vision and Pattern Recognition.Piscataway,NJ:IEEE,2017:2261-2269.
[11]TAN M X,LE Q V.Efficient Net:Rethinking Model Scaling for Convolutional Neural Networks[C]//36th International Confe-rence on Machine Learning(ICML).2019:10691-10700.
[12]CARLSSON G.Topology and Data[J].Bulletin of the American Mathematical Society,2009,46(2):255-308.
[13]OUDOT S Y.Persistence Theory:From Quiver Representations to Data Analysis[M].Providence:American Mathematical Society,2015.
[14]PUN C S,LEE S X,XIA K.Persistent-homology-based Machine Learning:A Survey and A Comparative Study[J].Artificial Intelligence Review,2022,55(7):5169-5213.
[15]ZHANG J L,JU X M.Application of Persistent Homology to Image Classification and Recognition[J].Chinese Journal of Communication on Applied Mathematics and Computation,2017,31(4):494-508.
[16]ZHANG Y J.Research on Digital Image Classification Based on Persistent Homology [D].Harbin:Harbin University of Commerce,2023.
[17]ZHANG Y P.Plant Leaf Image Classification by Using Persistent Homology and Multi-feature Fusion[D].Wuhan:Wuhan University of Technology,2021.
[18]ZHOU Z H.Application of Persistent Homology in Landmark Data's Statistical Analysis[D].Shenyang:Shenyang Normal University,2023.
[19]ALI D,ASAAD A,JIMENEZ M J,et al.A Survey of Vectorization Methods in Topological Data Analysis[J].IEEE Transactions on Pattern Analysis and Machine Intelligence,2023,45(12):14069-14080.
[20]WAGNER H,CHEN C,VUCINI E.Efficient Computation of Persistent Homology for Cubical Data[C]//Topological Me-thods in Data Analysis and Visualization II:Theory,Algorithms,and Applications.2011:91-106.
[21]EDELSBRUNNER H,HARER J.Computational Topology:An Introduction[M].American Mathematical Society,2009.
[22]周建伟.代数拓扑讲义[M].北京:科学出版社,2007.
[23]ALLEN H.Algebraic Topology[M].Cambridge:Cambridge University Press,2001.
[24]TURNER K,ROBINS V,MORGAN J.The Extended Persistent Homology Transform of Manifolds with Boundary[J].Journal of Applied and Computational Topology,2024,8(7):2111-2154.
[25]SAMVELYAN K,ZHANG J,POLTERO VICH L,et al.Topo-logical Persistence in Geometry and Analysis(University Lecture Series)[M].American Mathematical Society,2020.
[26]CRAWLEY-BOEVEY W.Decomposition of Pointwise Finite dimensional Persistence Modules[J].Journal of Algebra and Its Applications,2015,14(5):1550066.
[27]STUCKI N,PAETZOLD J C,SHIT S,et al.Topologically Faithful Image Segmentation via Induced Matching of Persis-tence Barcodes[C]//International Conference on Machine Learning.2023:32698-32727.
[28]FATEMA S,NUWAGIRA B,CHAKRABORTY S,et al.TopOC:Topological Deep Learning for Ovarian and Breast Cancer Diagnosis[C]//International Workshop on Topology and Graph-Informed Imaging Informatics.2024:22-32.
[29]CONTIF,MORONI D,PASCALI M A.A Topological Machine Learning Pipeline for Classification[J].Mathematics,2022,10(17):3086.
[30]MURRAY N,PERRONNIN F.Generalized Max Pooling[C]//Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition.2014:2473-2480.
[31]VASWANI A,SHAZEER N,PARMAR N,et al.Attention is All You Need[C]//31st Annual Conference on Neural Information Processing Systems(NIPS).2017:5999-6009.
[32]HAHNLOSER R H,SARPESHKAR R,MAH OWALD M A,et al.Digital Selection and Analogue Amplification Coexist in A Cortex-inspired Silicon Circuit[J].Nature,2000,405(6789):947-951.
[33]PANLAQUI B,DEB D.Convolutional Neural Networks:An Analysis of Varying Factors[C]//IEEE Southeast Conference.2023:348-349.
[34]XIAO H,RASUL K,VOLLGRAF R.Fashio-nmnist:A Novel Image Dataset for Benchmarking Machine Learning Algorithms [J].arXiv:1708.07747,2017.
[35]KRIZHEVSKY A,HINTON G.Learning Multiple Layers of Features from Tiny Images[J].Handbook of Systemic Autoimmune Diseases,2009,1(4).
[1] ZHANG Yan, ZHOU Jian, HAN Lei, CHENG Chunling. Gate-controlled Agent Attention Mechanism-based Soft Prompt Transfer Method [J]. Computer Science, 2026, 53(7): 139-145.
[2] DING Zhijun, AI Fangju, LIU Aihuan. Document-level Event Argument Extraction Model Based on Hierarchical Dependency Aggregation and Event Enhancement [J]. Computer Science, 2026, 53(7): 156-167.
[3] LI Kaiju, YIN Chenyang, CHENG Zhangtao, LIU Xueting, ZHOU Fan. Causal Subgraph Learning for Cascade Popularity Prediction [J]. Computer Science, 2026, 53(7): 213-221.
[4] XIAO Yanxue, DENG Li, REN Zhengwei, WU Mengxin. iDSRformer:Node Load Prediction Model for High-performance Computing Cluster [J]. Computer Science, 2026, 53(7): 251-261.
[5] DUAN Haiying, WANG Baohui, HUANG He. Malicious Traffic Detection Method of ICMP Covert Channel Based on Baseline Features [J]. Computer Science, 2026, 53(6A): 250200069-11.
[6] LI Jie, WANG Baohui, ZHANG Jingyuan. DDoS Attack Detection Based on Attention Mechanism TCN-BiLSTM [J]. Computer Science, 2026, 53(6A): 250300060-9.
[7] ZHANG Shouyi, SHEN Qiang, GUO Yiran, WANG Hanyu. Rain and Fog Weather Object Detection Algorithm Based on Improved YOLOv8 Model [J]. Computer Science, 2026, 53(6A): 250300090-7.
[8] YANG Geer, WANG Xin, SUN Wei, WANG Xinge, HU Zhongrui, MENG Wenjun, ZHANG Junqiang, WU Xinghui, LIU Jinshan, YAN Yuming. Survey on Positional Encoding Algorithms in Deep Learning [J]. Computer Science, 2026, 53(6A): 250300107-16.
[9] DUAN Pengsong, LUO Yu, WANG Chao. Q&A Model for Agricultural Diseases Based on Transformer [J]. Computer Science, 2026, 53(6A): 250400114-9.
[10] WEI Wei, LI Bicheng, ZHU Zhenshui, ZUO Jun. Semantic Modeling and Co-attention Mechanism for Multimodal Sarcasm Detection Method [J]. Computer Science, 2026, 53(6A): 250400127-6.
[11] FENG Guang, LIN Jianzhong, ZHONG Ting, ZHOU Yuanhua, ZHENG Runting, LIU Tianxiang. Triple Extraction Based on Pixel Difference Convolutional Network and Attention Mechanism [J]. Computer Science, 2026, 53(6A): 250400136-10.
[12] CHEN Dianlong, LIU Tengbin, GAO Xiong, TIAN Zijian, ZHU Wenbing, ZOU Shun, WANG Qiang. Defect Detection of Transmission Line Fittings Based on Multiscale Feature Fusion Attention and Cross-layer Aggregation [J]. Computer Science, 2026, 53(6A): 250600110-7.
[13] SHEN Yingchun, FENG Xiaohan, LI Qian. Accurate Recognition of Dialect Based on CTC-Conformer Model [J]. Computer Science, 2026, 53(6A): 250600112-8.
[14] ZHANG Zihao, WU Zezhong. Optimization of HAN-based GNN-Transformer Collaborative Contrastive Learning Framework [J]. Computer Science, 2026, 53(6A): 250900103-8.
[15] LI Siyu, QIAN Wenhua. HCKD:Lightweight Skin Lesion Classification Method Based on Dermoscopic Images [J]. Computer Science, 2026, 53(6A): 250600143-9.
Viewed
Full text


Abstract

Cited

  Shared   
  Discussed   
No Suggested Reading articles found!