Perception Encoder: The best visual embeddings are not at the output of the network

Daniel Bolya (Meta) · Po-Yao Huang (Meta) · Peize Sun (Meta) · Jang Hyun Cho (Meta Superintelligence Labs (MSL)) · Andrea Madotto (Meta Superintelligence Labs (MSL)) · Chen Wei (Johns Hopkins University) · Tengyu Ma (Meta) · Jiale Zhi (Meta) · Jathushan Rajasegaran (University of California Berkeley) · Hanoona Bangalath (Mohamed Bin Zayed University of Artificial Intelligence) · Junke Wang (Fudan University) · Marco Monteiro (Facebook) · Hu Xu (FAIR, Meta) · Shiyu Dong (Meta) · Nikhila Ravi (Facebook) · Shang-Wen Li (FAIR) · Piotr Dollar (Facebook AI Research) · Christoph Feichtenhofer (Meta)
alignment methodscontrastive vision-language trainingdense predictiondepth estimationdocument q&ageneral embeddingsintermediate layersmultimodal language modelingperception encoderpretraining objectivesspatial alignmentsynthetically annotated videostrackingvideo retrievalvision encoderszero-shot classification

We introduce Perception Encoder (PE), a family of state-of-the-art vision encoders for image and video understanding. Traditionally, vision encoders have relied on a variety of pretraining objectives, each excelling at different downstream tasks. Surprisingly, after scaling a carefully tuned image pretraining recipe and refining with a robust video data engine, we find that contrastive vision-language training alone can produce strong, general embeddings for all of these downstream tasks. There is only one caveat: these embeddings are hidden within the intermediate layers of the network. To draw them out, we introduce two alignment methods: language alignment for multimodal language modeling, and spatial alignment for dense prediction. Together, our PE family of models achieves state-of-the-art results on a wide variety of tasks, including zero-shot image and video classification and retrieval; document, image, and video Q&A; and spatial tasks such as detection, tracking, and depth estimation. We release our models, code, and novel dataset of synthetically and human-annotated videos: https://github.com/facebookresearch/perception_models