← Search

Mostafa Dehghani

31 accepted papers

2024

End-to-End Spatio-Temporal Action Localisation with Video Transformers

CVPR 2024poster

The most performant spatio-temporal action localisation models use external person proposals and complex external memory banks. We propose a fully end-to-end transformer based model that directly ingests an input video and outputs tubelets -- a sequence of bounding boxes and the action classes at ea…

Cited by 21SourcePDFScholar
2024

Fractal Patterns May Illuminate the Success of Next-Token Prediction

NeurIPS 2024poster

We study the fractal structure of language, aiming to provide a precise formalism for quantifying properties that may have been previously suspected but not formally shown. We establish that language is: (1) self-similar, exhibiting complexities at all levels of granularity, with no particular char…

Cited by 3SourcePDFScholar
2024

Frozen Feature Augmentation for Few-Shot Image Classification

CVPR 2024poster

Training a linear classifier or lightweight model on top of pretrained vision model outputs so-called 'frozen features' leads to impressive performance on a number of downstream few-shot tasks. Currently frozen features are not modified during training. On the other hand when networks are trained di…

Cited by 9SourcePDFScholar
2024

Low-Rank Adaptation for Multilingual Summarization: An Empirical Study

NAACL 2024findings

Although the advancements of pre-trained Large Language Models have significantly accelerated recent progress in NLP, their ever-increasing size poses significant challenges for conventional fine-tuning, especially in memory-intensive tasks. We investigate the potential of Parameter-Efficient Fine-T…

Cited by 4SourcePDFScholar
2024

On Scaling Up a Multilingual Vision and Language Model

CVPR 2024poster

We explore the boundaries of scaling up a multilingual vision and language model both in terms of size of the components and the breadth of its training task mixture. Our model achieves new levels of performance on a wide-range of varied and complex tasks including multiple image-based captioning an…

Cited by 8SourcePDFScholar
2023

$\Lambda$-DARTS: Mitigating Performance Collapse by Harmonizing Operation Selection among Cells

ICLR 2023poster

Differentiable neural architecture search (DARTS) is a popular method for neural architecture search (NAS), which performs cell-search and utilizes continuous relaxation to improve the search efficiency via gradient-based optimization. The main shortcoming of DARTS is performance collapse, where the…

Cited by 22SourcePDFScholar
2023

Adaptive Computation with Elastic Input Sequence

ICML 2023poster

Humans have the ability to adapt the type of information they use, the procedure they employ, and the amount of time they spend when solving problems. However, most standard neural networks have a fixed function type and computation budget regardless of the sample's nature or difficulty. Adaptivity…

2023

DSI++: Updating Transformer Memory with New Documents

EMNLP 2023long main

Differentiable Search Indices (DSIs) encode a corpus of documents in the parameters of a model and use the same model to map queries directly to relevant document identifiers. Despite the solid performance of DSI models, successfully deploying them in scenarios where document corpora change with tim…

Cited by 0SourceScholar
2023

Patch n’ Pack: NaViT, a Vision Transformer for any Aspect Ratio and Resolution

NeurIPS 2023poster

The ubiquitous and demonstrably suboptimal choice of resizing images to a fixed resolution before processing them with computer vision models has not yet been successfully challenged. However, models such as the Vision Transformer (ViT) offer flexible sequence-based modeling, and hence varying input…

Cited by 108SourcePDFScholar
2023

Scaling Laws vs Model Architectures: How does Inductive Bias Influence Scaling?

EMNLP 2023long findings

There have been a lot of interest in the scaling properties of Transformer models. However, not much has been done on the front of investigating the effect of scaling properties of different inductive biases and model architectures. Do model architectures scale differently? If so, how does inductive…

Cited by 0SourceScholar
2023

Scaling Vision Transformers to 22 Billion Parameters

ICML 2023oral

The scaling of Transformers has driven breakthrough capabilities for language models. At present, the largest large language models (LLMs) contain upwards of 100B parameters. Vision Transformers (ViT) have introduced the same architecture to image and video modelling, but these have not yet been suc…

Cited by 650SourcePDFScholar
2023

Sparse Upcycling: Training Mixture-of-Experts from Dense Checkpoints

ICLR 2023poster

Training large, deep neural networks to convergence can be prohibitively expensive. As a result, often only a small selection of popular, dense models are reused across different contexts and tasks. Increasingly, sparsely activated models, which seek to decouple model size from computation costs, ar…

2023

Transcending Scaling Laws with 0.1% Extra Compute

EMNLP 2023long main

Scaling language models improves performance but comes with significant computational costs. This paper proposes UL2R, a method that substantially improves existing language models and their scaling curves with a relatively tiny amount of extra compute. The key idea is to continue training a state-o…

Cited by 0SourceScholar
2023

UL2: Unifying Language Learning Paradigms

ICLR 2023poster

Existing pre-trained models are generally geared towards a particular class of problems. To date, there seems to be still no consensus on what the right architecture and pre-training setup should be. This paper presents a unified framework for pre-training models that are universally effective acros…

2022

Confident Adaptive Language Modeling

NeurIPS 2022accept

Recent advances in Transformer-based large language models (LLMs) have led to significant performance improvements across many tasks. These gains come with a drastic increase in the models' size, potentially leading to slow and costly use at inference time. In practice, however, the series of genera…

Cited by 227SourcePDFScholar
2022

Discrete Representations Strengthen Vision Transformer Robustness

ICLR 2022poster

Vision Transformer (ViT) is emerging as the state-of-the-art architecture for image recognition. While recent studies suggest that ViTs are more robust than their convolutional counterparts, our experiments find that ViTs are overly reliant on local features (\eg, nuisances and texture) and fail to…

Cited by 54SourcePDFScholar
2022

Exploring the Limits of Large Scale Pre-training

ICLR 2022spotlight

Recent developments in large-scale machine learning suggest that by scaling up data, model size and training time properly, one might observe that improvements in pre-training would transfer favorably to most downstream tasks. In this work we systematically study this phenomena and establish that,…

Cited by 138SourcePDFScholar
2022

Scale Efficiently: Insights from Pretraining and Finetuning Transformers

ICLR 2022poster

There remain many open questions pertaining to the scaling behaviour of Transformer architectures. These scaling decisions and findings can be critical, as training runs often come with an associated computational cost which have both financial and/or environmental impact. The goal of this paper is…

2022

Simple Open-Vocabulary Object Detection with Vision Transformers

ECCV 2022poster

"Combining simple architectures with large-scale pre-training has led to massive improvements in image classification. For object detection, pre-training and scaling approaches are less well established, especially in the long-tailed and open-vocabulary setting, where training data is relatively sca…

2022

Transformer Memory as a Differentiable Search Index

NeurIPS 2022accept

In this paper, we demonstrate that information retrieval can be accomplished with a single Transformer, in which all information about the corpus is encoded in the parameters of the model. To this end, we introduce the Differentiable Search Index (DSI), a new paradigm that learns a text-to-text mode…

Cited by 299SourcePDFScholar
2021

An Image is Worth 16x16 Words: Transformers for Image Recognition at Scale

ICLR 2021oral

While the Transformer architecture has become the de-facto standard for natural language processing tasks, its applications to computer vision remain limited. In vision, attention is either applied in conjunction with convolutional networks, or used to replace certain components of convolutional net…

2021

Are Pretrained Convolutions Better than Pretrained Transformers?

ACL 2021long

In the era of pre-trained language models, Transformers are the de facto choice of model architectures. While recent research has shown promise in entirely convolutional, or CNN, architectures, they have not been explored using the pre-train-fine-tune paradigm. In the context of language models, are…

2021

IDF++: Analyzing and Improving Integer Discrete Flows for Lossless Compression

ICLR 2021poster

In this paper we analyse and improve integer discrete flows for lossless compression. Integer discrete flows are a recently proposed class of models that learn invertible transformations for integer-valued random variables. Their discrete nature makes them particularly suitable for lossless compress…

Cited by 54SourcePDFScholar
2021

Long Range Arena : A Benchmark for Efficient Transformers

ICLR 2021poster

Transformers do not scale very well to long sequence lengths largely because of quadratic self-attention complexity. In the recent months, a wide spectrum of efficient, fast Transformers have been proposed to tackle this problem, more often than not claiming superior or comparable model quality to v…

2021

OmniNet: Omnidirectional Representations from Transformers

ICML 2021oral

This paper proposes Omnidirectional Representations from Transformers (OMNINET). In OmniNet, instead of maintaining a strictly horizon-tal receptive field, each token is allowed to attend to all tokens in the entire network. This process can also be interpreted as a form of extreme or intensive atte…

2021

Parameter-efficient Multi-task Fine-tuning for Transformers via Shared Hypernetworks

ACL 2021long

State-of-the-art parameter-efficient fine-tuning methods rely on introducing adapter modules between the layers of a pretrained language model. However, such modules are trained separately for each task and thus do not enable sharing information across tasks. In this paper, we show that we can learn…

2021

TokenLearner: Adaptive Space-Time Tokenization for Videos

NeurIPS 2021poster

In this paper, we introduce a novel visual representation learning which relies on a handful of adaptively learned tokens, and which is applicable to both image and video understanding tasks. Instead of relying on hand-designed splitting strategies to obtain visual tokens and processing a large numb…

Cited by 179SourcePDFScholar
2021

ViViT: A Video Vision Transformer

ICCV 2021poster

We present pure-transformer based models for video classification, drawing upon the recent success of such models in image classification. Our model extracts spatio-temporal tokens from the input video, which are then encoded by a series of transformer layers. In order to handle the long sequences o…

Cited by 2888PDFcodeScholar