用模型汤技术提升稀缺数据下非遗图像的分类准确率
Leveraging Model Soups to Classify Intangible Cultural Heritage Images from the Mekong Delta
- 用CoAtNet融合卷积与自注意力捕捉局部全局特征
- 在ICH-17数据集上达72.36%准确率,优于主流模型
- 通过几何多样性选择检查点,有效降低过拟合风险
湄公河三角洲非物质文化遗产(ICH)图像分类面临标注数据少、类别间视觉相似度高、领域异质性强等挑战。传统深度学习模型在低资源环境下易因方差大或学习虚假相关性而泛化能力差。本文提出一种结合CoAtNet与模型汤(model soups)的鲁棒框架,利用单次训练轨迹中多个检查点的轻量级加权平均,在不增加推理开销的前提下实现性能提升。CoAtNet通过分阶段融合卷积与自注意力机制捕捉多尺度特征。采用贪心与均匀两种集成策略,选择具有差异性的检查点构建最终模型。分析表明,模型汤能有效降低预测方差,仅引入微小偏差。通过交叉熵距离与多维缩放(MDS)验证,模型汤选取的是输出空间中几何分布多样化的模型,而非软投票那样聚集于中心区域的冗余模型。在包含7,406张图像、17个类别的ICH-17数据集上,本方法达到72.36%的top-1准确率和69.28%的宏F1分数,超越ResNet-50、DenseNet-121及ViT等强基线模型。结果证明,注重多样性的检查点平均是一种高效且原理清晰的降方差与增强泛化的方法,适用于文化丰富但数据稀缺的分类任务。
原文摘要 · Abstract (English)
The classification of Intangible Cultural Heritage (ICH) images in the Mekong Delta poses unique challenges due to limited annotated data, high visual similarity among classes, and domain heterogeneity. In such low-resource settings, conventional deep learning models often suffer from high variance or overfit to spurious correlations, leading to poor generalization. To address these limitations, we propose a robust framework that integrates the hybrid CoAtNet architecture with model soups, a lightweight weight-space ensembling technique that averages checkpoints from a single training trajectory without increasing inference cost. CoAtNet captures both local and global patterns through stage-wise fusion of convolution and self-attention. We apply two ensembling strategies - greedy and uniform soup - to selectively combine diverse checkpoints into a final model. Beyond performance improvements, we analyze the ensembling effect through the lens of bias-variance decomposition. Our findings show that model soups reduces variance by stabilizing predictions across diverse model snapshots, while introducing minimal additional bias. Furthermore, using cross-entropy-based distance metrics and Multidimensional Scaling (MDS), we show that model soups selects geometrically diverse checkpoints, unlike Soft Voting, which blends redundant models centered in output space. Evaluated on the ICH-17 dataset (7,406 images across 17 classes), our approach achieves state-of-the-art results with 72.36% top-1 accuracy and 69.28% macro F1-score, outperforming strong baselines including ResNet-50, DenseNet-121, and ViT. These results underscore that diversity-aware checkpoint averaging provides a principled and efficient way to reduce variance and enhance generalization in culturally rich, data-scarce classification tasks.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。