/EW-MTCNN-unofficial

Unofficial implementation of EW-MTCNN for image emotion classification

Primary LanguagePythonMIT LicenseMIT

EW-MTCNN-unofficial

Unofficial implementation of EW-MTCNN for image emotion classification

复现论文: 基于情感轮和多任务卷积神经网络的图像情感分布学习 (2022) 赖金水,万中英,曾雪强
这个论文甚至还有后续破事水 融合情感轮注意力的情感分布学习 (2023) 陈启凡,曾雪强,左家莉,万中英,王明文;或者看 CSDN博客解读

结论:能复现出论文数据,但情感轮先验层基本上卵用没用,严重拉低精度还不如没有 🤬

⚪ 模型结构

EM-MTCNN

⚪ 情感轮先验层权重

EW_layer

训练集精度 (我懒得做五折交叉了)

⚠ inplace eval on trainset :(

| -I | -M | cls_acc ↑ | sdl_kl ↓ | comment | | :-: | :-: | :-: | :-: | | none | - | 74.747% | 0.0456907 | 不使用情感轮先验层 | | original | freeze | 50.051% | 0.0912462 | 冻结情感轮先验层权重 | | original | unfreeze | 48.939% | 0.0741477 | 允许情感轮先验层参与微调 | | eye | unfreeze | 48.485% | 0.0774974 | 对角阵初始化 | | rand | unfreeze | 47.778% | 0.0732538 | 随机初始化 | | corr | freeze | 19.141% | 0.1249517 | | | corr | unfreeze | 27.677% | 0.1199178 | |

references


by Armit 2023/12/11