← Search

Jianmo Ni

15 accepted papers

2026

How to train data-efficient LLMs

ICLR 2026poster

The training of large language models (LLMs) is expensive. In this paper, we study data-efficient approaches for pre-training LLMs, \ie, techniques that aim to optimize the Pareto frontier of model quality and training resource/data consumption. We seek to understand the tradeoffs associated with da…

Cited by 0SourceScholar
2025

ActionPiece: Contextually Tokenizing Action Sequences for Generative Recommendation

ICML 2025spotlight

Generative recommendation (GR) is an emerging paradigm where user actions are tokenized into discrete token patterns and autoregressively generated as predictions. However, existing GR models tokenize each action independently, assigning the same fixed tokens to identical actions across all sequence…

2024

Leveraging LLMs for Synthesizing Training Data Across Many Languages in Multilingual Dense Retrieval

NAACL 2024long

There has been limited success for dense retrieval models in multilingual retrieval, due to uneven and scarce training data available across multiple languages. Synthetic training data generation is promising (e.g., InPars or Promptagator), but has been investigated only for English. Therefore, to s…

2023

A Suite of Generative Tasks for Multi-Level Multimodal Webpage Understanding

EMNLP 2023long main

Webpages have been a rich, scalable resource for vision-language and language only tasks. Yet only pieces of webpages are kept in existing datasets: image-caption pairs, long text articles, or raw HTML, never all in one place. Webpage tasks have resultingly received little attention and structured i…

Cited by 0SourcecodeScholar
2023

Promptagator: Few-shot Dense Retrieval From 8 Examples

ICLR 2023poster

Much recent research on information retrieval has focused on how to transfer from one task (typically with abundant supervised data) to various other retrieval tasks where supervision is limited, with the implicit assumption that it is possible to generalize from one task to all the rest. However, t…

Cited by 230SourcePDFScholar
2022

ExT5: Towards Extreme Multi-Task Scaling for Transfer Learning

ICLR 2022poster

Despite the recent success of multi-task learning and transfer learning for natural language processing (NLP), few works have systematically studied the effect of scaling up the number of tasks during pre-training. Towards this goal, this paper introduces ExMix (Extreme Mixture): a massive collectio…

Cited by 222SourcePDFScholar
2022

Exploring Dual Encoder Architectures for Question Answering

EMNLP 2022main

Dual encoders have been used for question-answering (QA) and information retrieval (IR) tasks with good results. There are two major types of dual encoders, Siamese Dual Encoders (SDE), with parameters shared across two encoders, and Asymmetric Dual Encoder (ADE), with two distinctly parameterized e…

Cited by 20SourcePDFScholar
2022

Large Dual Encoders Are Generalizable Retrievers

EMNLP 2022main

It has been shown that dual encoders trained on one domain often fail to generalize to other domains for retrieval tasks. One widespread belief is that the bottleneck layer of a dual encoder, where the final score is simply a dot-product between a query vector and a passage vector, is too limited co…

2022

LongT5: Efficient Text-To-Text Transformer for Long Sequences

NAACL 2022findings

Recent work has shown that either (1) increasing the input length or (2) increasing model size can improve the performance of Transformer-based neural models. In this paper, we present LongT5, a new model that explores the effects of scaling both the input length and model size at the same time. Spe…

2022

SHARE: a System for Hierarchical Assistive Recipe Editing

EMNLP 2022main

The large population of home cooks with dietary restrictions is under-served by existing cooking resources and recipe generation models. To help them, we propose the task of controllable recipe editing: adapt a base recipe to satisfy a user-specified dietary constraint. This task is challenging, and…

2022

Sentence-T5: Scalable Sentence Encoders from Pre-trained Text-to-Text Models

ACL 2022findings

We provide the first exploration of sentence embeddings from text-to-text transformers (T5) including the effects of scaling up sentence encoders to 11B parameters. Sentence embeddings are broadly useful for language processing tasks. While T5 achieves impressive performance on language tasks, it is…

2022

Transformer Memory as a Differentiable Search Index

NeurIPS 2022accept

In this paper, we demonstrate that information retrieval can be accomplished with a single Transformer, in which all information about the corpus is encoded in the parameters of the model. To this end, we introduce the Differentiable Search Index (DSI), a new paradigm that learns a text-to-text mode…

Cited by 299SourcePDFScholar
2021

Multi-stage Training with Improved Negative Contrast for Neural Passage Retrieval

EMNLP 2021main

In the context of neural passage retrieval, we study three promising techniques: synthetic data generation, negative sampling, and fusion. We systematically investigate how these techniques contribute to the performance of the retrieval system and how they complement each other. We propose a multi-s…