用视觉Transformer和Mamba融合模型提升人脸美感预测准确率
VM-BeautyNet: A Synergistic Ensemble of Vision Transformer and Mamba for Facial Beauty Prediction
- 双模型协同:ViT抓全局结构,Mamba高效处理长程依赖
- 在SCUT-FBP5500上达到PC 0.9212、MAE 0.2085、RMSE 0.2698
- 可视化验证互补性,适合计算美学与多模态模型研究者
人脸美感预测(FBP)是复杂且具有挑战性的计算机视觉任务,旨在建模人类审美感知的主观性和复杂性。尽管深度学习模型,尤其是卷积神经网络(CNN),已取得显著进展,但通常难以捕捉对人类判断至关重要的全局、整体面部特征。视觉变压器(ViT)通过有效建模长距离空间关系加以弥补,但其二次复杂度成为瓶颈。本文提出一种新型异构集成架构——VM-BeautyNet,协同融合视觉变压器与基于Mamba的视觉模型(状态空间模型最新进展)。ViT骨干擅长捕捉全局面部结构与对称性,而Mamba骨干以线性复杂度高效建模长程依赖,聚焦序列特征与纹理。我们在基准数据集SCUT-FBP5500上评估该方法,所提VM-BeautyNet达到最先进性能:皮尔逊相关系数(PC)为0.9212,平均绝对误差(MAE)为0.2085,均方根误差(RMSE)为0.2698。此外,通过Grad-CAM可视化分析,我们证实了两个骨干网络的互补特征提取能力,为模型决策过程提供新见解,并展示了一种强大的计算美学新架构范式。
原文摘要 · Abstract (English)
Facial Beauty Prediction (FBP) is a complex and challenging computer vision task, aiming to model the subjective and intricate nature of human aesthetic perception. While deep learning models, particularly Convolutional Neural Networks (CNNs), have made significant strides, they often struggle to capture the global, holistic facial features that are critical to human judgment. Vision Transformers (ViT) address this by effectively modeling long-range spatial relationships, but their quadratic complexity can be a bottleneck. This paper introduces a novel, heterogeneous ensemble architecture, \textbf{VM-BeautyNet}, that synergistically fuses the complementary strengths of a Vision Transformer and a Mamba-based Vision model, a recent advancement in State-Space Models (SSMs). The ViT backbone excels at capturing global facial structure and symmetry, while the Mamba backbone efficiently models long-range dependencies with linear complexity, focusing on sequential features and textures. We evaluate our approach on the benchmark SCUT-FBP5500 dataset. Our proposed VM-BeautyNet achieves state-of-the-art performance, with a \textbf{Pearson Correlation (PC) of 0.9212}, a \textbf{Mean Absolute Error (MAE) of 0.2085}, and a \textbf{Root Mean Square Error (RMSE) of 0.2698}. Furthermore, through Grad-CAM visualizations, we provide interpretability analysis that confirms the complementary feature extraction of the two backbones, offering new insights into the model's decision-making process and presenting a powerful new architectural paradigm for computational aesthetics.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。