TensorX
返回文献探索

Paper · arXiv 2308.01544

Multimodal Neurons in Pretrained Text-Only Transformers

Sarah Schwettmann, Neil Chowdhury, Antonio Torralba

17 upvotesAugust 3, 2023arXiv 预印本
AI 摘要

Frozen text transformers can incorporate visual information through self-supervised learning, with specific neurons facilitating translation between modalities and influencing image captioning.

text transformerself-supervised visual encoderlinear projectionmultimodal neuronsresidual streamimage captioning

Abstract

Language models demonstrate remarkable capacity to generalize representations learned in one modality to downstream tasks in other modalities. Can we trace this ability to individual neurons? We study the case where a frozen text transformer is augmented with vision using a self-supervised visual encoder and a single linear projection learned on an image-to-text task. Outputs of the projection layer are not immediately decodable into language describing image content; instead, we find that translation between modalities occurs deeper within the transformer. We introduce a procedure for identifying "multimodal neurons" that convert visual representations into corresponding text, and decoding the concepts they inject into the model's residual stream. In a series of experiments, we show that multimodal neurons operate on specific visual concepts across inputs, and have a systematic causal effect on image captioning.

北京市昌平区探索星信息技术及软件开发工作室

京ICP备2026059466号