Timezone: »

 
Poster
Adapting Self-Supervised Vision Transformers by Probing Attention-Conditioned Masking Consistency
Viraj Prabhu · Sriram Yenamandra · Aaditya Singh · Judy Hoffman

Tue Nov 29 09:00 AM -- 11:00 AM (PST) @ Hall J #214

Visual domain adaptation (DA) seeks to transfer trained models to unseen, unlabeled domains across distribution shift, but approaches typically focus on adapting convolutional neural network architectures initialized with supervised ImageNet representations. In this work, we shift focus to adapting modern architectures for object recognition -- the increasingly popular Vision Transformer (ViT) -- initialized with modern pretraining based on self-supervised learning (SSL). Inspired by the design of recent SSL approaches based on learning from partial image inputs generated via masking or cropping -- either by learning to predict the missing pixels, or learning representational invariances to such augmentations -- we propose PACMAC, a two-stage adaptation algorithm for self-supervised ViTs. PACMAC first performs in-domain SSL on pooled source and target data to learn task-discriminative features, and then probes the model's predictive consistency across a set of partial target inputs generated via a novel attention-conditioned masking strategy, to identify reliable candidates for self-training. Our simple approach leads to consistent performance gains over competing methods that use ViTs and self-supervised initializations on standard object recognition benchmarks. Our code is available at https://github.com/virajprabhu/PACMAC.

Author Information

Viraj Prabhu (Georgia Tech)
Viraj Prabhu

I am a fourth year CS Ph.D. student at Georgia Tech, advised by Judy Hoffman. My research interests are in developing data-efficient and reliable computer vision systems that can be deployed in the real world. Specifically, I am interested in sample-efficient learning (particularly few-shot and active learning), adaptation across visual tasks and domains, and reliable and calibrated uncertainty estimation from deep neural networks.

Sriram Yenamandra (Georgia Tech)
Aaditya Singh (Georgia Institute of Technology)
Aaditya Singh

I am a Master's student in Computer Science at Georgia Tech advised by Judy Hoffman. Recently, I interned at AWS AI in the Computer Vision Science team. Earlier, I worked at IBM Research and graduated from the Indian Institute of Technology Kanpur with a major in Electrical Engineering. I am broadly interested in Self-Supervised and Multi-Modal Learning.

Judy Hoffman (Georgia Institute of Technology)

More from the Same Authors

  • 2022 : Augmentation Consistency-guided Self-training for Source-free Domain Adaptive Semantic Segmentation »
    Viraj Prabhu · Shivam Khare · Deeksha Kartik · Judy Hoffman
  • 2022 : Bi-Directional Self-Attention for Vision Transformers »
    George Stoica · Taylor Hearn · Bhavika Devnani · Judy Hoffman
  • 2022 Poster: ZSON: Zero-Shot Object-Goal Navigation using Multimodal Goal Embeddings »
    Arjun Majumdar · Gunjan Aggarwal · Bhavika Devnani · Judy Hoffman · Dhruv Batra
  • 2021 : Panel: Future directions for tackling distribution shifts »
    Tatsunori Hashimoto · Jamie Morgenstern · Judy Hoffman · Andrew Beck
  • 2020 Poster: Auxiliary Task Reweighting for Minimum-data Learning »
    Baifeng Shi · Judy Hoffman · Kate Saenko · Trevor Darrell · Huijuan Xu
  • 2019 : Poster Session I »
    Shuangjia Zheng · Arnav Kapur · Umar Asif · Eyal Rozenberg · Cyprien Gilet · Oleksii Sidorov · Yogesh Kumar · Tom Van Steenkiste · William Boag · David Ouyang · Paul Jaeger · Sheng Liu · Aparna Balagopalan · Deepta Rajan · Marta Skreta · Nikhil Pattisapu · Jann Goschenhofer · Viraj Prabhu · Di Jin · Laura-Jayne Gardiner · Irene Li · sriram kumar · Qiyuan Hu · Mehul Motani · Justin Lovelace · Usman Roshan · Lucy Lu Wang · Ilya Valmianski · Hyeonwoo Lee · Sunil Mallya · Elias Chaibub Neto · Jonas Kemp · Marie Charpignon · Amber Nigam · Wei-Hung Weng · Sabri Boughorbel · Alexis Bellot · Lovedeep Gondara · Haoran Zhang · Taha Bahadori · John Zech · Rulin Shao · Edward Choi · Laleh Seyyed-Kalantari · Emily Aiken · Ioana Bica · Yiqiu Shen · Kieran Chin-Cheong · Subhrajit Roy · Ioana Baldini · So Yeon Min · Dirk Deschrijver · Pekka Marttinen · Damian Pascual Ortiz · Supriya Nagesh · Niklas Rindtorff · Andriy Mulyar · Katharina Hoebel · Martha Shaka · Pierre Machart · Leon Gatys · Nathan Ng · Matthias Hüser · Devin Taylor · Dennis Barbour · Natalia Martinez · Clara McCreery · Benjamin Eyre · Vivek Natarajan · Ren Yi · Ruibin Ma · Chirag Nagpal · Nan Du · Chufan Gao · Anup Tuladhar · Sam Shleifer · Jason Ren · Pouria Mashouri · Ming Yang Lu · Farideh Bagherzadeh-Khiabani · Olivia Choudhury · Maithra Raghu · Scott Fleming · Mika Jain · GUO YANG · Alena Harley · Stephen Pfohl · Elisabeth Rumetshofer · Alex Fedorov · Saloni Dash · Jacob Pfau · Sabina Tomkins · Colin Targonski · Michael Brudno · Xinyu Li · Yiyang Yu · Nisarg Patel
  • 2018 : Poster session: Contributed papers »
    Michael Cvitkovic · Arijit Patra · Yunpeng Li · RAHMAN BANYA SAFF SANYA · Guanghua Chi · Benjamin Huynh · Hamed Alemohammad · Simón Ramírez Amaya · Nazmus Saquib · Jade Abbott · Teo de Campos · Viraj Prabhu · Alvaro Riascos · Hafte Abera · praney dubey · Tanushyam Chattopadhyay · Hsiang Hsu · Mayank Jain · Kartikeya Bhardwaj · Gabriel Cadamuro · Bradley Gram-Hansen · Georg Dorffner
  • 2018 : Poster Session I »
    Aniruddh Raghu · Daniel Jarrett · Kathleen Lewis · Elias Chaibub Neto · Nicholas Mastronarde · Shazia Akbar · Chun-Hung Chao · Henghui Zhu · Seth Stafford · Luna Zhang · Jen-Tang Lu · Changhee Lee · Adityanarayanan Radhakrishnan · Fabian Falck · Liyue Shen · Daniel Neil · Yusuf Roohani · Aparna Balagopalan · Brett Marinelli · Hagai Rossman · Sven Giesselbach · Jose Javier Gonzalez Ortiz · Edward De Brouwer · Byung-Hoon Kim · Rafid Mahmood · Tzu Ming Hsu · Antonio Ribeiro · Rumi Chunara · Agni Orfanoudaki · Kristen Severson · Mingjie Mai · Sonali Parbhoo · Albert Haque · Viraj Prabhu · Di Jin · Alena Harley · Geoffroy Dubourg-Felonneau · Xiaodan Hu · Maithra Raghu · Jonathan Warrell · Nelson Johansen · Wenyuan Li · Marko Järvenpää · Satya Narayan Shukla · Sarah Tan · Vincent Fortuin · Beau Norgeot · Yi-Te Hsu · Joel H Saltz · Veronica Tozzo · Andrew Miller · Guillaume Ausset · Azin Asgarian · Francesco Paolo Casale · Antoine Neuraz · Bhanu Pratap Singh Rawat · Turgay Ayer · Xinyu Li · Mehul Motani · Nathaniel Braman · Laetitia M Shao · Adrian Dalca · Hyunkwang Lee · Emma Pierson · Sandesh Ghimire · Yuji Kawai · Owen Lahav · Anna Goldenberg · Denny Wu · Pavitra Krishnaswamy · Colin Pawlowski · Arijit Ukil · Yuhui Zhang