Unofficial implementation of EW-MTCNN for image emotion classification
复现论文: 基于情感轮和多任务卷积神经网络的图像情感分布学习 (2022) 赖金水,万中英,曾雪强
这个论文甚至还有后续破事水 融合情感轮注意力的情感分布学习 (2023) 陈启凡,曾雪强,左家莉,万中英,王明文;或者看 CSDN博客解读
结论:能复现出论文数据,但情感轮先验层基本上卵用没用,严重拉低精度还不如没有 🤬
⚪ 模型结构
⚪ 情感轮先验层权重
⚪ 训练集精度 (我懒得做五折交叉了)
⚠ 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 | |
- 论文
- Emotion6 数据集
by Armit 2023/12/11