Demo Data from EDBT 2026 / reviewers in the wild / expert
Neel Jain
dblp:277/1176
· DBLP profile ↗
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
| Topic | Weight | Papers | Last | Evidence papers |
|---|---|---|---|---|
Natural language and speech › Language models and text generation › large language model evaluation
benchmark contamination |
0.9 | 1 | 2025 | LiveBench: A Challenging, Contamination-Limited LLM Benchmark · ICLR 2025 |
Natural language and speech › Language models and text generation
large language model evaluation |
0.9 | 1 | 2025 | 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.9 | 1 | 2025 | 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.9 | 1 | 2025 | 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.9 | 1 | 2025 | 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.8 | 1 | 2024 | NEFTune: Noisy Embeddings Improve Instruction Finetuning · ICLR 2024 |
Natural language and speech › Language models and text generation › compositional generalization
length generalization |
0.8 | 1 | 2024 | Transformers Can Do Arithmetic with the Right Embeddings · NeurIPS 2024 |
Machine learning › Generative modeling › diffusion model
memorization mitigation |
0.8 | 1 | 2024 | 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.8 | 1 | 2024 | Transformers Can Do Arithmetic with the Right Embeddings · NeurIPS 2024 |
Machine learning › Deep learning architectures and training
positional encoding |
0.8 | 1 | 2024 | Transformers Can Do Arithmetic with the Right Embeddings · NeurIPS 2024 |
Distributed systems › distributed machine learning
distributed training |
0.8 | 1 | 2024 | Democratizing AI: Open-source Scalable LLM Training on GPU-based Supercomputers · SC 2024 |
High-performance computing
large-scale training |
0.8 | 1 | 2024 | 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.7 | 1 | 2023 | 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.7 | 1 | 2023 | 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.7 | 1 | 2023 | Hard Prompts Made Easy: Gradient-Based Discrete Optimization for Prompt Tuning and Discovery · NeurIPS 2023 |
Machine learning › Reinforcement learning
benchmark design |
0.3 | 1 | 2025 | LiveBench: A Challenging, Contamination-Limited LLM Benchmark · ICLR 2025 |
Natural language and speech › Language models and text generation
large language model training |
0.2 | 1 | 2024 | 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.2 | 1 | 2024 | 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
| Year | Publication | Venue | Position |
|---|---|---|---|
| 2025 | LiveBench: A Challenging, Contamination-Limited LLM BenchmarkabstractTest 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 |
ICLR | 8 |
| 2025 | Scaling up Test-Time Compute with Latent Reasoning: A Recurrent Depth ApproachabstractWe 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 |
NeurIPS | 3 |
| 2024 | NEFTune: Noisy Embeddings Improve Instruction FinetuningabstractWe 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 |
ICLR | 1 |
| 2024 | Be like a Goldfish, Don't Memorize! Mitigating Memorization in Generative LLMsabstractLarge 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 |
NeurIPS | 4 |
| 2024 | Transformers Can Do Arithmetic with the Right EmbeddingsabstractThe 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 |
NeurIPS | 4 |
| 2024 | Democratizing AI: Open-source Scalable LLM Training on GPU-based SupercomputersabstractTraining 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 |
SC | 7 |
| 2023 | Hard Prompts Made Easy: Gradient-Based Discrete Optimization for Prompt Tuning and DiscoveryabstractThe 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 |
NeurIPS | 2 |