ML Anthology

Rethinking Attention: Exploring Shallow Feed-Forward Neural Networks as an Alternative to Attention Layers in Transformers (Student Abstract)

Danilo Dordevic, Vukasin Bozic, Joseph Thommes, Daniele Coppola, Sidak Pal Singh
AAAI 2024 pp. 23477-23479
doi:10.1609/AAAI.V38I21.30436 /aaai/2024/dordevic2024aaai-rethinking/

Abstract

This work presents an analysis of the effectiveness of using standard shallow feed-forward networks to mimic the behavior of the attention mechanism in the original Transformer model, a state-of-the-art architecture for sequence-to-sequence tasks. We substitute key elements of the attention mechanism in the Transformer with simple feed-forward networks, trained using the original components via knowledge distillation. Our experiments, conducted on the IWSLT2017 dataset, reveal the capacity of these ”attentionless Transformers” to rival the performance of the original architecture. Through rigorous ablation studies, and experimenting with various replacement network types and sizes, we offer insights that support the viability of our approach. This not only sheds light on the adaptability of shallow feed-forward networks in emulating attention mechanisms but also underscores their potential to streamline complex architectures for sequence-to-sequence tasks.

PDF AAAI Semantic Scholar

Cite

Text

Dordevic et al. "Rethinking Attention: Exploring Shallow Feed-Forward Neural Networks as an Alternative to Attention Layers in Transformers (Student Abstract)." AAAI Conference on Artificial Intelligence, 2024. doi:10.1609/AAAI.V38I21.30436

Markdown

[Dordevic et al. "Rethinking Attention: Exploring Shallow Feed-Forward Neural Networks as an Alternative to Attention Layers in Transformers (Student Abstract)." AAAI Conference on Artificial Intelligence, 2024.](https://mlanthology.org/aaai/2024/dordevic2024aaai-rethinking/) doi:10.1609/AAAI.V38I21.30436

BibTeX

@inproceedings{dordevic2024aaai-rethinking,
  title     = {{Rethinking Attention: Exploring Shallow Feed-Forward Neural Networks as an Alternative to Attention Layers in Transformers (Student Abstract)}},
  author    = {Dordevic, Danilo and Bozic, Vukasin and Thommes, Joseph and Coppola, Daniele and Singh, Sidak Pal},
  booktitle = {AAAI Conference on Artificial Intelligence},
  year      = {2024},
  pages     = {23477-23479},
  doi       = {10.1609/AAAI.V38I21.30436},
  url       = {https://mlanthology.org/aaai/2024/dordevic2024aaai-rethinking/}
}