Demonstration venue · read-only. Every page can be browsed; the buttons that would change it are switched off. Create an account to run TaxoReview on your own data. A simulation from public publication records: the people listed were not invited and did not agree to review.
? Tutorial for this page

Neel Jain

dblp:277/1176 · DBLP profile ↗
← Back
7ranked-venue papers
1as first author
7since 2021 · last 2025
—ORCID · unresolved

Domains — the database's venue-derived domains; a paper can count in several

Artificial intelligence and machine learning · 6 · 1 first-author · 6 since 2021Systems, architecture and hardware · 1 · 1 since 2021

Expertise — from the expertise taxonomy: the topics of the expert's papers under the CCF categories. A weight counts papers with recency: 1 for a paper about the topic, 0.3 when the topic is its context, halved every five years.

Artificial intelligence
7 papers
Language models and text generation · 70% Deep learning architectures and training · 14% Generative modeling · 12%
Computer architecture, parallel and distributed computing, and storage systems
1 paper
High-performance computing · 50% Distributed systems · 50%

Topics — the 18 heaviest of 19, each with the papers that count most for it

TopicWeightPapersLastEvidence papers
Natural language and speech › Language models and text generation › large language model evaluation
benchmark contamination
0.912025
LiveBench: A Challenging, Contamination-Limited LLM Benchmark · ICLR 2025
Natural language and speech › Language models and text generation
large language model evaluation
0.912025
LiveBench: A Challenging, Contamination-Limited LLM Benchmark · ICLR 2025
Natural language and speech › Language models and text generation › large language model reasoning › inference-time reasoning
latent reasoning
0.912025
Scaling up Test-Time Compute with Latent Reasoning: A Recurrent Depth Approach · NeurIPS 2025
Machine learning › Deep learning architectures and training
recurrent neural network
0.912025
Scaling up Test-Time Compute with Latent Reasoning: A Recurrent Depth Approach · NeurIPS 2025
Natural language and speech › Language models and text generation
test-time scaling
0.912025
Scaling up Test-Time Compute with Latent Reasoning: A Recurrent Depth Approach · NeurIPS 2025
Natural language and speech › Language models and text generation
instruction tuning
0.812024
NEFTune: Noisy Embeddings Improve Instruction Finetuning · ICLR 2024
Natural language and speech › Language models and text generation › compositional generalization
length generalization
0.812024
Transformers Can Do Arithmetic with the Right Embeddings · NeurIPS 2024
Machine learning › Generative modeling › diffusion model
memorization mitigation
0.812024
Be like a Goldfish, Don't Memorize! Mitigating Memorization in Generative LLMs · NeurIPS 2024
Natural language and speech › Language models and text generation › mathematical reasoning
numerical reasoning
0.812024
Transformers Can Do Arithmetic with the Right Embeddings · NeurIPS 2024
Machine learning › Deep learning architectures and training
positional encoding
0.812024
Transformers Can Do Arithmetic with the Right Embeddings · NeurIPS 2024
Distributed systems › distributed machine learning
distributed training
0.812024
Democratizing AI: Open-source Scalable LLM Training on GPU-based Supercomputers · SC 2024
High-performance computing
large-scale training
0.812024
Democratizing AI: Open-source Scalable LLM Training on GPU-based Supercomputers · SC 2024
Natural language and speech › Language models and text generation › prompting › prompt engineering
prompt optimization
0.712023
Hard Prompts Made Easy: Gradient-Based Discrete Optimization for Prompt Tuning and Discovery · NeurIPS 2023
Natural language and speech › Language models and text generation
prompt tuning
0.712023
Hard Prompts Made Easy: Gradient-Based Discrete Optimization for Prompt Tuning and Discovery · NeurIPS 2023
Machine learning › Generative modeling › diffusion model
text-to-image generation
0.712023
Hard Prompts Made Easy: Gradient-Based Discrete Optimization for Prompt Tuning and Discovery · NeurIPS 2023
Machine learning › Reinforcement learning
benchmark design
0.312025
LiveBench: A Challenging, Contamination-Limited LLM Benchmark · ICLR 2025
Natural language and speech › Language models and text generation
large language model training
0.212024
Democratizing AI: Open-source Scalable LLM Training on GPU-based Supercomputers · SC 2024
Machine learning › Efficient and distributed learning › distributed training › parallelization
parallel training
0.212024
Democratizing AI: Open-source Scalable LLM Training on GPU-based Supercomputers · SC 2024

Methods — techniques the papers use, named apart from their topics

speculative decoding · 0.9recurrent depth · 0.9crowdsourcing · 0.9automated scoring · 0.9positional embedding · 0.8performance modeling · 0.8next-token training objective · 0.8instruction fine-tuning · 0.8input injection · 0.8goldfish loss · 0.8four-dimensional hybrid parallelism · 0.8embedding noise · 0.8
YearPublicationVenuePosition
2025 LiveBench: A Challenging, Contamination-Limited LLM Benchmark
abstract
Test set contamination, wherein test data from a benchmark ends up in a newer model's training set, is a well-documented obstacle for fair LLM evaluation and can quickly render benchmarks obsolete. To mitigate this, many recent benchmarks crowdsource new prompts and evaluations from human or LLM judges; however, these can introduce significant biases, and break down when scoring hard questions. In this work, we introduce a new benchmark for LLMs designed to be resistant to both test set contamination and the pitfalls of LLM judging and human crowdsourcing. We release LiveBench, the first benchmark that (1) contains frequently-updated questions from recent information sources, (2) scores answers automatically according to objective ground-truth values, and (3) contains a wide variety of challenging tasks, spanning math, coding, reasoning, language, instruction following, and data analysis. To achieve this, LiveBench contains questions that are based on recently-released math competitions, arXiv papers, news articles, and datasets, and it contains harder, contamination-limited versions of tasks from previous benchmarks such as Big-Bench Hard, AMPS, and IFEval. We evaluate many prominent closed-source models, as well as dozens of open-source models ranging from 0.5B to 405B in size. LiveBench is difficult, with top models achieving below 70% accuracy. We release all questions, code, and model answers. Questions are added and updated on a monthly basis, and we release new tasks and harder versions of tasks over time so that LiveBench can distinguish between the capabilities of LLMs as they improve in the future. We welcome community engagement and collaboration for expanding the benchmark tasks and models.
Colin White, Samuel Dooley, Manley Roberts, Arka Pal, Benjamin Feuer, Siddhartha Jain 0001, Ravid Shwartz-Ziv, Neel Jain, Khalid Saifullah, Sreemanti Dey, Shubh-Agrawal, Sandeep Singh Sandha, Siddartha Naidu, Chinmay Hegde, Yann LeCun, Tom Goldstein, Willie Neiswanger, Micah Goldblum
ICLR8
2025 Scaling up Test-Time Compute with Latent Reasoning: A Recurrent Depth Approach
abstract
We study a novel language model architecture that is capable of scaling test-time computation by implicitly reasoning in latent space. Our model works by iterating a recurrent block, thereby unrolling to arbitrary depth at test-time. This stands in contrast to mainstream reasoning models that scale up compute by producing more tokens. Unlike approaches based on chain-of-thought, our approach does not require any specialized training data, can work with small context windows, and can capture types of reasoning that are not easily represented in words. We train a proof-of-concept model from scratch with 3.5 billion parameters and 800 billion tokens. We show that this model can effortlessly use varying levels of compute, significantly improving with additional compute especially on reasoning tasks, such as math and coding. Further, this architecture naturally reduces compute costs via zero-shot per-token adaptive compute, KV-cache sharing and speculative decoding.
Jonas Geiping, Sean McLeish, Neel Jain, John Kirchenbauer, Brian R. Bartoldson, Bhavya Kailkhura, Abhinav Bhatele, Tom Goldstein
NeurIPS3
2024 NEFTune: Noisy Embeddings Improve Instruction Finetuning
abstract
We show that language model finetuning can be improved, sometimes dramatically, with a simple augmentation. NEFTune adds noise to the embedding vectors during training. Standard finetuning of LLaMA-2-7B using Alpaca achieves $29.79$\% on AlpacaEval, which rises to $64.69$\% using noisy embeddings. NEFTune also improves over strong baselines on modern instruction datasets. Models trained with Evol-Instruct see a $10$\% improvement, with ShareGPT an $8$\% improvement, and with OpenPlatypus an $8$\% improvement. Even powerful models further refined with RLHF such as LLaMA-2-Chat benefit from additional training with NEFTune. Particularly, we see these improvements on the conversational abilities of the instruction model and not on traditional tasks like those on the OpenLLM Leaderboard, where performance is the same.
Neel Jain, Ping-Yeh Chiang, Yuxin Wen, John Kirchenbauer, Hong-Min Chu, Gowthami Somepalli, Brian R. Bartoldson, Bhavya Kailkhura, Avi Schwarzschild, Aniruddha Saha, Micah Goldblum, Jonas Geiping, Tom Goldstein
ICLR1
2024 Be like a Goldfish, Don't Memorize! Mitigating Memorization in Generative LLMs
abstract
Large language models can memorize and repeat their training data, causing privacy and copyright risks. To mitigate memorization, we introduce a subtle modification to the next-token training objective that we call the goldfish loss. During training, a randomly sampled subsets of tokens are excluded from the loss computation. These dropped tokens are not memorized by the model, which prevents verbatim reproduction of a complete chain of tokens from the training set. We run extensive experiments training billion-scale LLaMA-2 models, both pre-trained and trained from scratch, and demonstrate significant reductions in extractable memorization with little to no impact on downstream benchmarks. _Code and checkpoints: https://github.com/ahans30/goldfish-loss_
Abhimanyu Hans, John Kirchenbauer, Yuxin Wen, Neel Jain, Hamid Kazemi, Prajwal Singhania, Gowthami Somepalli, Jonas Geiping, Abhinav Bhatele, Tom Goldstein
NeurIPS4
2024 Transformers Can Do Arithmetic with the Right Embeddings
abstract
The poor performance of transformers on arithmetic tasks seems to stem in large part from their inability to keep track of the exact position of each digit inside of a large span of digits. We mend this problem by adding an embedding to each digit that encodes its position relative to the start of the number. In addition to the boost these embeddings provide on their own, we show that this fix enables architectural modifications such as input injection and recurrent layers to improve performance even further. With positions resolved, we can study the logical extrapolation ability of transformers. Can they solve arithmetic problems that are larger and more complex than those in their training data? We find that training on only 20 digit numbers with a single GPU for one day, we can reach state-of-the-art performance, achieving up to 99% accuracy on 100 digit addition problems. Finally, we show that these gains in numeracy also unlock improvements on other multi-step reasoning tasks including sorting and multiplication.
Sean McLeish, Arpit Bansal, Alex Stein, Neel Jain, John Kirchenbauer, Brian R. Bartoldson, Bhavya Kailkhura, Abhinav Bhatele, Jonas Geiping, Avi Schwarzschild, Tom Goldstein
NeurIPS4
2024 Democratizing AI: Open-source Scalable LLM Training on GPU-based Supercomputers
abstract
Training and fine-tuning large language models (LLMs) with hundreds of billions to trillions of parameters requires tens of thousands of GPUs, and a highly scalable software stack. In this work, we present a novel four-dimensional hybrid parallel algorithm implemented in a highly scalable, portable, open-source framework called AxoNn. We describe several performance optimizations in AxoNN to improve matrix multiply kernel performance, overlap non-blocking collectives with computation, and performance modeling to choose performance optimal configurations. These have resulted in unprecedented scaling and peak flop/s (bf16) for training of GPT-style transformer models on Perlmutter (620.1 Petaflop/s), Frontier (1.381 Exaflop/s) and Alps (1.423 Exaflop/s). While the abilities of LLMs improve with the number of trainable parameters, so do privacy and copyright risks caused by memorization of training data, which can cause disclosure of sensitive or private information at inference time. We highlight this side effect of scale through experiments that explore “catastrophic memorization,” where models are sufficiently large to memorize training data in a single pass, and present an approach to prevent it. As part of this study, we demonstrate fine-tuning of a 405-billion parameter LLM using AxoNN on Frontier.
Prajwal Singhania, Aditya K. Ranjan, John Kirchenbauer, Jonas Geiping, Yuxin Wen, Neel Jain, Abhimanyu Hans, Manli Shu, Aditya Tomar, Tom Goldstein, Abhinav Bhatele
SC7
2023 Hard Prompts Made Easy: Gradient-Based Discrete Optimization for Prompt Tuning and Discovery
abstract
The strength of modern generative models lies in their ability to be controlled through prompts. Hard prompts comprise interpretable words and tokens, and are typically hand-crafted by humans. Soft prompts, on the other hand, consist of continuous feature vectors. These can be discovered using powerful optimization methods, but they cannot be easily edited, re-used across models, or plugged into a text-based interface. We describe an easy-to-use approach to automatically optimize hard text prompts through efficient gradient-based optimization. Our approach can be readily applied to text-to-image and text-only applications alike. This method allows API users to easily generate, discover, and mix and match image concepts without prior knowledge of how to prompt the model. Furthermore, using our method, we can bypass token-level content filters imposed by Midjourney by optimizing through the open-sourced text encoder.
Yuxin Wen, Neel Jain, John Kirchenbauer, Micah Goldblum, Jonas Geiping, Tom Goldstein
NeurIPS2