A Frozen Pretrained Transformer

Home

Discover

Bookmarks

Keep Up With Latest Trending Papers. Computer Science, AI and more.Subscribe

Added to collectionTop 100 Papers By Signal Trends

CC BY

Source arXiv

Machine Learning

Artificial Intelligence

A Frozen Pretrained Transformer

Pretrained Transformers as Universal Computation Engines

We investigate the capability of a transformer pretrained on natural language to generalize to other modalities with minimal finetuning, in particular, without finetuning of the self-attention and feedforward layers of the residual blocks.

We consider such a model, which we call a frozen pretrained transformer(fpt), and study finetuning it on a variety of sequence classification tasks spanning numerical computation, vision, and protein fold prediction.

In contrast to prior works which investigate finetuning on the same modality as the pretraining dataset, we show that pretraining on natural language improves performance and compute efficiency on non-language downstream tasks.

In particular, we find that such pretraining enables such a model to generalize in zero-shot to these modalities, matching the performance of a transformer fully trained on these tasks.

Authors

Kevin Lu, Aditya Grover, Pieter Abbeel, Igor Mordatch

◐ Recommended Follows

◐ Latest Activity