activity
20202024
most citedLearning Neural Network Subspaces

15 citations · 66 across the 17 of their papers we have counts for

collaborators

18 papers

cs.AI2024

The Llama 3 Herd of Models

Aaron Grattafiori, Abhimanyu Dubey, Abhinav Jauhri +556

Modern artificial intelligence (AI) systems are powered by foundation models. This paper presents a new set of foundation models, called Llama 3. It is a herd of language models th…

cs.DC2024

KV-Runahead: Scalable Causal LLM Inference by Parallel Key-Value Cache Generation

Minsik Cho, Mohammad Rastegari, Devang Naik

Large Language Model or LLM inference has two phases, the prompt (or prefill) phase to output the first token and the extension (or decoding) phase to the generate subsequent token…

cs.LG2023★ 2 cited

Weight subcloning: direct initialization of transformers using larger pretrained ones

Mohammad Samragh, Mehrdad Farajtabar, Sachin Mehta +5

Training large transformer models from scratch for a target task requires lots of data and is computationally demanding. The usual practice of transfer learning overcomes this chal…

cs.CL2023★ 13 cited

LLM in a flash: Efficient Large Language Model Inference with Limited Memory

Keivan Alizadeh, Iman Mirzadeh, Dmitry Belenko +5

Large language models (LLMs) are central to modern natural language processing, delivering exceptional performance in various tasks. However, their substantial computational and me…

cs.CV2023

Knowledge Transfer from Vision Foundation Models for Efficient Training of Small Task-specific Models

Raviteja Vemulapalli, Hadi Pouransari, Fartash Faghri +4

Vision Foundation Models (VFMs) pretrained on massive datasets exhibit impressive performance on various downstream tasks, especially with limited labeled target data. However, due…

cs.LG2023★ 7 cited

ReLU Strikes Back: Exploiting Activation Sparsity in Large Language Models

Iman Mirzadeh, Keivan Alizadeh, Sachin Mehta +5

Large Language Models (LLMs) with billions of parameters have drastically transformed AI applications. However, their demanding computation during inference has raised significant…

cs.CV2023★ 13 cited

SAM-CLIP: Merging Vision Foundation Models towards Semantic and Spatial Understanding

Haoxiang Wang, Pavan Kumar Anasosalu Vasu, Fartash Faghri +6

The landscape of publicly available vision foundation models (VFMs), such as CLIP and Segment Anything Model (SAM), is expanding rapidly. VFMs are endowed with distinct capabilitie…

cs.SD2023

Diffusion Models as Masked Audio-Video Learners

Elvis Nunez, Yanzi Jin, Mohammad Rastegari +2

Over the past several years, the synchronization between audio and visual signals has been leveraged to learn richer audio-visual representations. Aided by the large availability o…

cs.CL2023

Do Compressed LLMs Forget Knowledge? An Experimental Study with Practical Implications

Duc N. M Hoang, Minsik Cho, Thomas Merth +2

Compressing Large Language Models (LLMs) often leads to reduced performance, especially for knowledge-intensive tasks. In this work, we dive into how compression damages LLMs' inhe…

cs.LG2023

eDKM: An Efficient and Accurate Train-time Weight Clustering for Large Language Models

Minsik Cho, Keivan A. Vahid, Qichen Fu +5

Since Large Language Models or LLMs have demonstrated high-quality performance on many complex language tasks, there is a great interest in bringing these LLMs to mobile devices fo…

cs.CV2023

On the Efficacy of Multi-scale Data Samplers for Vision Applications

Elvis Nunez, Thomas Merth, Anish Prabhu +4

Multi-scale resolution training has seen an increased adoption across multiple vision tasks, including classification and detection. Training with smaller resolutions enables faste…

cs.CV2023★ 3 cited

Bytes Are All You Need: Transformers Operating Directly On File Bytes

Maxwell Horton, Sachin Mehta, Ali Farhadi +1

Modern deep learning approaches usually utilize modality-specific processing. For example, the most common deep learning approach to image classification involves decoding image fi…

cs.CV2023

Reinforce Data, Multiply Impact: Improved Model Accuracy and Robustness with Dataset Reinforcement

Fartash Faghri, Hadi Pouransari, Sachin Mehta +4

We propose Dataset Reinforcement, a strategy to improve a dataset once such that the accuracy of any model architecture trained on the reinforced dataset is improved at no addition…

cs.CV2022★ 1 cited

RangeAugment: Efficient Online Augmentation with Range Learning

Sachin Mehta, Saeid Naderiparizi, Fartash Faghri +5

State-of-the-art automatic augmentation methods (e.g., AutoAugment and RandAugment) for visual recognition tasks diversify training data using a large set of augmentation operation…

cs.CV2022

SPIN: An Empirical Evaluation on Sharing Parameters of Isotropic Networks

Chien-Yu Lin, Anish Prabhu, Thomas Merth +4

Recent isotropic networks, such as ConvMixer and vision transformers, have found significant success across visual recognition tasks, matching or outperforming non-isotropic convol…

cs.LG2021★ 15 cited

Learning Neural Network Subspaces

Mitchell Wortsman, Maxwell Horton, Carlos Guestrin +2

Recent observations have advanced our understanding of the neural network optimization landscape, revealing the existence of (1) paths of high accuracy containing diverse solutions…

cs.CV2020★ 4 cited

Layer-Wise Data-Free CNN Compression

Maxwell Horton, Yanzi Jin, Ali Farhadi +1

We present a computationally efficient method for compressing a trained neural network without using real data. We break the problem of data-free network compression into independe…

cs.LG2020★ 8 cited

Supermasks in Superposition

Mitchell Wortsman, Vivek Ramanujan, Rosanne Liu +4

We present the Supermasks in Superposition (SupSup) model, capable of sequentially learning thousands of tasks without catastrophic forgetting. Our approach uses a randomly initial…