← Search

Yifeng Lu

12 accepted papers

2026

Rethinking generative image pretraining: How far are we from scaling up next-pixel prediction?

ICML 2026poster

This paper investigates the scaling properties of autoregressive next-pixel prediction, a simple, end-to-end yet under-explored framework for unified vision models. Starting with images at resolutions of 32x32, we train a family of Transformers using IsoFlops profiles across compute budgets up to 7e…

Cited by 0SourceScholar
2024

Large Language Models as Optimizers

ICLR 2024poster

Optimization is ubiquitous. While derivative-based algorithms have been powerful tools for various problems, the absence of gradient imposes challenges on many real-world applications. In this work, we propose Optimization by PROmpting (OPRO), a simple and effective approach to leverage large langua…

2024

Long-form factuality in large language models

NeurIPS 2024poster

Large language models (LLMs) often generate content that contains factual errors when responding to fact-seeking prompts on open-ended topics. To benchmark a model’s long-form factuality in open domains, we first use GPT-4 to generate LongFact, a prompt set comprising thousands of questions spanning…

2023

Brainformers: Trading Simplicity for Efficiency

ICML 2023poster

Transformers are central to recent successes in natural language processing and computer vision. Transformers have a mostly uniform backbone where layers alternate between feed-forward and self-attention in order to build a deep network. Here we investigate this design choice and find that more comp…

Cited by 37SourcePDFScholar
2023

DoReMi: Optimizing Data Mixtures Speeds Up Language Model Pretraining

NeurIPS 2023spotlight

The mixture proportions of pretraining data domains (e.g., Wikipedia, books, web text) greatly affect language model (LM) performance. In this paper, we propose Domain Reweighting with Minimax Optimization (DoReMi), which first trains a small proxy model using group distributionally robust optimizat…

2023

Symbol tuning improves in-context learning in language models

EMNLP 2023long main

We present symbol tuning - finetuning language models on in-context input-label pairs where natural language labels (e.g., "positive/negative sentiment") are replaced with arbitrary symbols (e.g., "foo/bar"). Symbol tuning leverages the intuition that when a model cannot use instructions or natural…

Cited by 0SourceScholar
2023

Symbolic Discovery of Optimization Algorithms

NeurIPS 2023poster

We present a method to formulate algorithm discovery as program search, and apply it to discover optimization algorithms for deep neural network training. We leverage efficient search techniques to explore an infinite and sparse program space. To bridge the large generalization gap between proxy and…

Cited by 518SourcePDFScholar
2022

DeepFusion: Lidar-Camera Deep Fusion for Multi-Modal 3D Object Detection

CVPR 2022poster

Lidars and cameras are critical sensors that provide complementary information for 3D detection in autonomous driving. While prevalent multi-modal methods simply decorate raw lidar point clouds with camera features and feed them directly to existing 3D detection models, our study shows that fusing c…

Cited by 476PDFcodeScholar
2022

TabNAS: Rejection Sampling for Neural Architecture Search on Tabular Datasets

NeurIPS 2022accept

The best neural architecture for a given machine learning problem depends on many factors: not only the complexity and structure of the dataset, but also on resource constraints including latency, compute, energy consumption, etc. Neural architecture search (NAS) for tabular datasets is an important…

2020

PyGlove: Symbolic Programming for Automated Machine Learning

NeurIPS 2020oral

Neural networks are sensitive to hyper-parameter and architecture choices. Automated Machine Learning (AutoML) is a promising paradigm for automating these choices. Current ML software libraries, however, are quite limited in handling the dynamic interactions among the components of AutoML. For exam…