← Search

Aaron Gokaslan

25 accepted papers

2026

OpenThoughts: Data Recipes for Reasoning Models

ICLR 2026oral

Reasoning models have made rapid progress on many benchmarks involving math, code, and science. Yet, there are still many open questions about the best train- ing recipes for reasoning since state-of-the-art models often rely on proprietary datasets with little to no public information available. To…

Cited by 0SourcecodeScholar
2025

Block Diffusion: Interpolating Between Autoregressive and Diffusion Language Models

ICLR 2025oral

Diffusion language models offer unique benefits over autoregressive models due to their potential for parallelized generation and controllability, yet they lag in likelihood modeling and are limited to fixed-length generation. In this work, we introduce a class of block diffusion language models tha…

2025

Encoder-Decoder Diffusion Language Models for Efficient Training and Inference

NeurIPS 2025poster

Discrete diffusion models enable parallel token sampling for faster inference than autoregressive approaches. However, prior diffusion models use a decoder-only architecture, which requires sampling algorithms that invoke the full network at every denoising step and incur high computational cost. Ou…

Cited by 0SourceScholar
2025

Position: Benchmarking is Broken - Don't Let AI be Its Own Judge

NeurIPS 2025poster

The meteoric rise of Artificial Intelligence (AI), with its rapidly expanding market capitalization, presents both transformative opportunities and critical challenges. Chief among these is the urgent need for a new, unified paradigm for trustworthy evaluation, as current benchmarks increasingly rev…

Cited by 0SourceScholar
2025

The Common Pile v0.1: An 8TB Dataset of Public Domain and Openly Licensed Text

NeurIPS 2025poster

Large language models (LLMs) are typically trained on enormous quantities of unlicensed text, a practice that has led to scrutiny due to possible intellectual property infringement and ethical concerns. Training LLMs on openly licensed text presents a first step towards addressing these issues, but…

Cited by 0SourceScholar
2025

The Diffusion Duality

ICML 2025poster

Uniform-state discrete diffusion models hold the promise of fast text generation due to their inherent ability to self-correct. However, they are typically outperformed by autoregressive models and masked diffusion models. In this work, we narrow this performance gap by leveraging a key insight: Un…

2024

Caduceus: Bi-Directional Equivariant Long-Range DNA Sequence Modeling

ICML 2024poster

Large-scale sequence modeling has sparked rapid advances that now extend into biology and genomics. However, modeling genomic sequences introduces challenges such as the need to model long-range token interactions, the effects of upstream and downstream regions of the genome, and the reverse complem…

2024

CommonCanvas: Open Diffusion Models Trained on Creative-Commons Images

CVPR 2024poster

We train a set of open text-to-image (T2I) diffusion models on a dataset of curated Creative-Commons-licensed (CC) images which yields models that are competitive with Stable Diffusion 2 (SD2). This task presents two challenges: (1) high-resolution CC images lack the captions necessary to train T2I…

Cited by 30SourcePDFScholar
2024

DataComp-LM: In search of the next generation of training sets for language models

NeurIPS 2024poster

We introduce DataComp for Language Models, a testbed for controlled dataset experiments with the goal of improving language models. As part of DCLM, we provide a standardized corpus of 240T tokens extracted from Common Crawl, effective pretraining recipes based on the OpenLM framework, and a broad s…

Cited by 64SourcePDFScholar
2024

Diffusion Models With Learned Adaptive Noise

NeurIPS 2024spotlight

Diffusion models have gained traction as powerful algorithms for synthesizing high-quality images. Central to these algorithms is the diffusion process, a set of equations which maps data to noise in a way that can significantly affect performance. In this paper, we explore whether the diffusion p…

2024

Position: Standardization of Behavioral Use Clauses is Necessary for the Adoption of Responsible Licensing of AI

ICML 2024poster

Growing concerns over negligent or malicious uses of AI have increased the appetite for tools that help manage the risks of the technology. In 2018, licenses with behaviorial-use clauses (commonly referred to as Responsible AI Licenses) were proposed to give developers a framework for releasing AI a…

Cited by 0SourcePDFScholar
2024

Simple and Effective Masked Diffusion Language Models

NeurIPS 2024poster

While diffusion models excel at generating high-quality images, prior work reports a significant performance gap between diffusion and autoregressive (AR) methods in language modeling. In this work, we show that simple masked discrete diffusion is more performant than previously thought. We apply an…

2024

The GAN is dead; long live the GAN! A Modern GAN Baseline

NeurIPS 2024poster

There is a widely-spread claim that GANs are difficult to train, and GAN architectures in the literature are littered with empirical tricks. We provide evidence against this claim and build a modern GAN baseline in a more principled manner. First, we derive a well-behaved regularized relativistic GA…

2023

Galactic: Scaling End-to-End Reinforcement Learning for Rearrangement at 100k Steps-per-Second

CVPR 2023poster

We present Galactic, a large-scale simulation and reinforcement-learning (RL) framework for robotic mobile manipulation in indoor environments. Specifically, a Fetch robot (equipped with a mobile base, 7DoF arm, RGBD camera, egomotion, and onboard sensing) is spawned in a home environment and asked…

2023

Habitat-Matterport 3D Semantics Dataset

CVPR 2023highlight

We present the Habitat-Matterport 3D Semantics (HM3DSEM) dataset. HM3DSEM is the largest dataset of 3D real-world spaces with densely annotated semantics that is currently available to the academic community. It consists of 142,646 object instance annotations across 216 3D spaces and 3,100 rooms wit…

2023

InfoDiffusion: Representation Learning Using Information Maximizing Diffusion Models

ICML 2023poster

While diffusion models excel at generating high-quality samples, their latent variables typically lack semantic meaning and are not suitable for representation learning. Here, we propose InfoDiffusion, an algorithm that augments diffusion models with low-dimensional latent variables that capture hig…

Cited by 41SourcePDFScholar
2022

The BigScience ROOTS Corpus: A 1.6TB Composite Multilingual Dataset

NeurIPS 2022accept

As language models grow ever larger, the need for large-scale high-quality text datasets has never been more pressing, especially in multilingual settings. The BigScience workshop, a 1-year international and multidisciplinary initiative, was formed with the goal of researching and training large lan…

Cited by 214SourcePDFScholar
2021

Habitat 2.0: Training Home Assistants to Rearrange their Habitat

NeurIPS 2021spotlight

We introduce Habitat 2.0 (H2.0), a simulation platform for training virtual robots in interactive 3D environments and complex physics-enabled scenarios. We make comprehensive contributions to all levels of the embodied AI stack – data, simulation, and benchmark tasks. Specifically, we present: (i) R…

2021

Habitat-Matterport 3D Dataset (HM3D): 1000 Large-scale 3D Environments for Embodied AI

NeurIPS 2021poster

We present the Habitat-Matterport 3D (HM3D) dataset. HM3D is a large-scale dataset of 1,000 building-scale 3D reconstructions from a diverse set of real-world locations. Each scene in the dataset consists of a textured 3D mesh reconstruction of interiors such as multi-floor residences, stores, and ot…

Cited by 434SourcecodeScholar
2021

THDA: Treasure Hunt Data Augmentation for Semantic Navigation

ICCV 2021poster

Can general-purpose neural models learn to navigate? For PointGoal navigation (""go to x, y""), the answer is a clear `yes' -- mapless neural models composed of task-agnostic components (CNNs and RNNs) trained with large-scale model-free reinforcement learning achieve near-perfect performance. Howev…

Cited by 91PDFScholar
2021

TöRF: Time-of-Flight Radiance Fields for Dynamic Scene View Synthesis

NeurIPS 2021poster

Neural networks can represent and accurately reconstruct radiance fields for static 3D scenes (e.g., NeRF). Several works extend these to dynamic scenes captured with monocular video, with promising performance. However, the monocular setting is known to be an under-constrained problem, and so metho…

2021

Waypoint Models for Instruction-Guided Navigation in Continuous Environments

ICCV 2021poster

Little inquiry has explicitly addressed the role of action spaces in language-guided visual navigation -- either in terms of its effect on navigation success or the efficiency with which a robotic agent could execute the resulting trajectory. Building on the recently released VLN-CE setting for inst…

Cited by 94PDFcodeScholar
2020

MatryODShka: Real-time 6DoF Video View Synthesis using Multi-Sphere Images

ECCV 2020poster

We introduce a method to convert stereo 360 (omnidirectional stereo) imagery into a layered, multi-sphere image representation for six degree-of-freedom (6DoF) rendering. Stereo 360 imagery can be captured from multi-camera systems for virtual reality (VR) rendering, but lacks motion parallax and co…

2020

Sim2Real Predictivity: Does Evaluation in Simulation Predict Real-World Performance?

RA-L 2020

Does progress in simulation translate to progress on robots? If one method outperforms another in simulation, how likely is that trend to hold in reality on a robot? We examine this question for embodied PointGoal navigation - developing engineering tools and a research paradigm for evaluating a sim

Cited by 252SourceScholar
2018

Improving Shape Deformation in Unsupervised Image-to-Image Translation

ECCV 2018poster

Unsupervised image-to-image translation techniques are able to map local texture between two domains, but they are typically un- successful when the domains require larger shape change. Inspired by semantic segmentation, we introduce a discriminator with dilated convo- lutions which is able to use i…