Reweighting Framewise Attention in Video Transformers for Facial Expression Understanding

· Source: cs.CV updates on arXiv.org · Field: Technology & Digital — Artificial Intelligence & Machine Learning, Computer Vision · Depth: Expert, extended

Summary

MiRA (Marginal-induced Attention Redistribution) is a novel plug-in framework designed to enhance Vision Transformer (ViT) backbones for understanding subtle facial expressions in videos. It addresses the limitation of standard ViT models, which often prioritize dominant global motions over fine-grained facial dynamics. MiRA achieves this by deriving frame-level confidence and intra-frame concentration statistics from self-attention maps, then redistributing attention towards localized facial cues without adding trainable parameters. The framework offers two modes: an "exact mode" for principled post-softmax redistribution and a "flashLite mode," an efficient pre-softmax approximation compatible with FlashAttention kernels. Experimental results demonstrate MiRA's consistent improvements on challenging Facial Expression Recognition (FER) benchmarks, including DFEW, MAFW, and FERV39k, outperforming several general video models. The flashLite mode further provides a 20% latency reduction and 26% throughput gain on an H100 GPU, making it highly efficient.

Key takeaway

For Machine Learning Engineers developing video-based facial expression recognition systems, integrating MiRA into your Vision Transformer backbones can significantly improve performance. This plug-in module enhances sensitivity to subtle facial dynamics by intelligently redistributing attention, outperforming existing general video models. Consider deploying the flashLite mode for its computational efficiency, offering a 20% latency reduction and 26% throughput gain on H100 GPUs, especially when scaling to larger models or real-world applications without task-specific preprocessing.

Key insights

Reweighting ViT attention based on frame-level confidence and intra-frame concentration improves subtle facial dynamics understanding.

Principles

Method

MiRA computes frame-level confidence and intra-frame concentration from attention maps (exact) or key energies (flashLite), combines them into a marginal importance score, and uses this to redistribute attention via scaling factors or logit biases.

In practice

Topics

Code references

Best for: Research Scientist, AI Scientist, Machine Learning Engineer, Computer Vision Engineer

Related on AIssential

Open in AIssential →

Editorial summary, takeaway, and curation by AIssential. Original article published by cs.CV updates on arXiv.org.