Is this you? Claim this profile to correct it, add a bio and choose the work people see first.

Claim this profile

Works102 from public data

TitleCited by
  • On the Opportunities and Risks of Foundation Models

    Rishi Bommasani, Drew A. Hudson, Ehsan Adeli, Russ Biagio Altman, Simran Arora, Sydney von Arx, Michael S. Bernstein, Jeannette Bohg, +22 more

    arXiv · 2021

    2,227
  • Mobility network models of COVID-19 explain inequities and inform reopening

    Serina Y. Chang, Emma Pierson, Pang Wei Koh, Jaline Gerardin, Beth Redbird, David B. Grusky, Jure Leskovec

    Nature · 2020

    A metapopulation susceptible–exposed–infectious–removed (SEIR) model that integrates fine-grained, dynamic mobility networks to simulate the spread of SARS-CoV-2 in ten of the largest US metropolitan areas is introduced and correctly predicts higher infection rates among disadvantaged racial and socioeconomic groups.

    1,561
  • 1,322
  • Distributionally Robust Neural Networks

    Shiori Sagawa, Pang Wei Koh, Tatsunori Hashimoto, Percy Liang

    International Conference on Learning Representations · 2020

    The results suggest that regularization is critical for worst-group performance in the overparameterized regime, even if it is not needed for average performance, and introduce and provide convergence guarantees for a stochastic optimizer for this group DRO setting.

    807
  • DataComp: In search of the next generation of multimodal datasets

    Samir Yitzhak Gadre, Gabriel Ilharco, Fang, Alex, Jonathan Hayase, Georgios Smyrnis, Thao D. Nguyen, Ryan Marten, Mitchell Wortsman, +22 more

    arXiv · 2023

    The introduction of DataComp, a testbed for dataset experiments centered around a new candidate pool of 12.8 billion image-text pairs from Common Crawl, enables training a CLIP ViT-L/14 from scratch to 79.2% zero-shot accuracy on ImageNet, and makes the benchmark accessible to researchers with varying resources.

    782
  • OpenFlamingo: An Open-Source Framework for Training Large Autoregressive Vision-Language Models

    Anas Awadalla, Irena Gao, Josh Gardner, Jack Hessel, Yusuf Hanafy, Wanrong Zhu, Kalyani Marathe, Yonatan Bitton, +8 more

    arXiv · 2023

    OpenFlamingo is an ongoing effort to produce an open-source replication of DeepMind's Flamingo models, and the models, training data, hyperparameters, and evaluation suite are introduced.

    663
  • Mapping the Pairwise Choices Leading from Pluripotency to Human Bone, Heart, and Other Mesoderm Cell Types

    Kyle M. Loh, Angela Chen, Pang Wei Koh, Tianda Z. Deng, Rahul Sinha, Jonathan M. Tsai, Amira A. Barkal, Kimberle Shen, +15 more

    Cell · 2016

    The extrinsic signals controlling each binary lineage decision were defined, enabling us to logically block differentiation towards unwanted fates and rapidly steer pluripotent stem cells towards 80–99% pure human mesodermal lineages at most branchpoints.

    490
  • On Random Weights and Unsupervised Feature Learning

    Andrew Saxe, Pang Wei Koh, Zhenghao Chen, Maneesh Bhand, Bipin Suresh, Andrew Y. Ng

    International Conference on Machine Learning · 2011

    The answer is that certain convolutional pooling architectures can be inherently frequency selective and translation invariant, even with random weights, and the viability of extremely fast architecture search is demonstrated by using random weights to evaluate candidate architectures, thereby sidestepping the time-consuming learning process.

    467
  • Are aligned neural networks adversarially aligned?

    Nicholas Carlini, Milad Nasr, Christopher A. Choquette-Choo, Matthew Jagielski, Irena Gao, Anas Awadalla, Pang Wei Koh, Daphne Ippolito, +3 more

    arXiv · 2023

    It is shown that existing NLP-based optimization attacks are insufficiently powerful to reliably attack aligned text models: even when current NLP-based attacks fail, the authors can find adversarial inputs with brute force.

    415
  • FActScore: Fine-grained Atomic Evaluation of Factual Precision in Long Form Text Generation

    Sewon Min, Kalpesh Krishna, Xinxi Lyu, Mike Lewis, Wen-tau Yih, Pang Wei Koh, Mohit Iyyer, Luke Zettlemoyer, +1 more

    Conference on Empirical Methods in Natural Language Processing (EMNLP) · 2023

    372
  • Tiled convolutional neural networks

    Jiquan Ngiam, Zhenghao Chen, Daniel Chia, Pang Wei Koh, Quoc Viet Le, Andrew Y. Ng

    neural information processing systems · 2010

    This paper proposes tiled convolution neural networks (Tiled CNNs), which use a regular "tiled" pattern of tied weights that does not require that adjacent hidden units share identical weights, but instead requires only that hidden units k steps away from each other to have tied weights.

    353
  • OLMoE: Open Mixture-of-Experts Language Models

    Niklas Muennighoff, Luca Soldaini, Dirk Groeneveld, Kyle Lo, Jacob Morrison, Sewon Min, Weijia Shi, Walsh, Pete, +16 more

    arXiv · 2024

    This work introduces OLMoE, a fully open, state-of-the-art language model leveraging sparse Mixture-of-Experts (MoE), and presents various experiments on MoE training, analyzes routing in the model showing high specialization, and open-source all aspects of the work: model weights, training data, code, and logs.

    352
  • 349
  • Toward Trustworthy AI Development: Mechanisms for Supporting Verifiable Claims

    Miles Brundage, Shahar Avin, Jasmine Wang, Haydn Belfield, Gretchen Marie Krueger, Gillian K. Hadfield, Heidy Khlaaf, Jingying Yang, +22 more

    arXiv · 2020

    296
  • Stronger data poisoning attacks break data sanitization defenses

    Pang Wei Koh, Jacob Steinhardt, Percy Liang

    Machine Learning · 2021

    Three attacks are developed that can bypass a broad range of common data sanitization defenses, including anomaly detectors based on nearest neighbors, training loss, and singular-value decomposition, and the Karush–Kuhn–Tucker conditions.

    292
  • 2 OLMo 2 Furious

    Team OLMo, Walsh, Pete, Luca Soldaini, Dirk Groeneveld, Kyle Lo, Shane Arora, Akshita Bhagia, Yuling Gu, +22 more

    arXiv · 2024

    This work describes the modified model architecture and training recipe, focusing on techniques for achieving better training stability and improved per-token efficiency, and introduces a new, specialized data mix called Dolmino Mix 1124, which significantly improves model capabilities across many downstream task benchmarks when introduced via late-stage curriculum training.

    286
  • Sparse Filtering

    Jiquan Ngiam, Zhenghao Chen, Sonia A. Bhaskar, Pang Wei Koh, Andrew Y. Ng

    neural information processing systems · 2011

    This work presents sparse filtering, a simple new algorithm which is efficient and only has one hyperparameter, the number of features to learn, and evaluates it on natural images, object classification, and phone classification, showing that the method works well on a range of different modalities.

    278
  • WILDS: A Benchmark of in-the-Wild Distribution Shifts

    Pang Wei Koh, Shiori Sagawa, Henrik Marklund, Sang Michael Xie, Marvin Zhang, Akshay Balsubramani, Weihua Hu, Michihiro Yasunaga, +15 more

    CaltechAUTHORS (California Institute of Technology) · 2020

    269
  • Spurious Rewards: Rethinking Training Signals in RLVR

    Rulin Shao, Shuyue Stella Li, Rui Xin, Scott Geng, Yiping Wang, Sewoong Oh, Simon S. Du, Nathan Lambert, +6 more

    arXiv · 2025

    235
  • MediQ: Question-Asking LLMs and a Benchmark for Reliable Interactive Clinical Reasoning

    Shuyue Li, Vidhisha Balachandran, Shangbin Feng, Jonathan Seth Ilgen, Emma Pierson, Pang Wei Koh, Yulia Tsvetkov

    neural information processing systems · 2024

    This paper proposes to change the static paradigm to an interactive one, develop systems that proactively ask questions to gather more information and respond reliably, and introduces an interactive MediQ benchmark and a novel question-asking system to evaluate question-asking ability in LLMs.

    218
  • Learning Deep Energy Models

    Jiquan Ngiam, Zhenghao Chen, Pang Wei Koh, Andrew Y. Ng

    International Conference on Machine Learning · 2011

    This work proposes deep energy models, which use deep feedforward neural networks to model the energy landscapes that define probabilistic models, and is able to efficiently train all layers of this model simultaneously, allowing the lower layers of the model to adapt to the training of the higher layers, and thereby producing better generative models.

    204
  • Impossibility theorems for feature attribution

    Blair Bilodeau, Natasha Jaques, Pang Wei Koh, Been Kim

    Proceedings of the National Academy of Sciences · 2024

    This work shows that complete and linear methods are provably less reliable than simpler methods at answering basic feature attribution questions, and provides impossibility results that highlight their failure cases.

    167
  • Concept Bottleneck Models

    Pang Wei Koh, Thao N. N. Nguyen, Yew Siang Tang, Stephen Mussmann, Emma Pierson, Been Kim, Percy Liang

    arXiv · 2020

    144
  • Wild-Time: A Benchmark of in-the-Wild Distribution Shift over Time

    Huaxiu Yao, Choi, Caroline, Bochuan Cao, Yoonho Lee, Pang Wei Koh, Chelsea Finn

    arXiv · 2022

    Wild-Time is curate, a benchmark of 5 datasets that reflect temporal distribution shifts arising in a variety of real-world applications, including patient prognosis and news classification that systematically benchmark 13 prior approaches, including methods in domain generalization, continual learning, self-supervised learning, and ensemble learning.

    143
  • The Generative AI Paradox: "What It Can Create, It May Not Understand"

    West, Peter, Ximing Lu, Nouha Dziri, Faeze Brahman, Linjie Li, Jena D. Hwang, Liwei Jiang, Jillian Fisher, +6 more

    arXiv · 2023

    The hypothesis that models' generative capability may not be contingent upon understanding capability, and call for caution in interpreting artificial intelligence by analogy to human intelligence, is supported.

    136
  • Synthesizing scientific literature with retrieval-augmented language models

    Akari Asai, Jacqueline He, Rulin Shao, Weijia Shi, Amanpreet Singh, Joseph Chee Chang, Kyle Lo, Luca Soldaini, +20 more

    Nature · 2026

    A specialized, open-source, retrieval-augmented language model is introduced for answering scientific queries and synthesizing literature, the responses of which are shown to be preferred by human evaluations over expert-written answers.

    105
  • An Investigation of Why Overparameterization Exacerbates Spurious Correlations

    Shiori Sagawa, Aditi Raghunathan, Pang Wei Koh, Percy Liang

    arXiv · 2020

    105
  • Reliable, Adaptable, and Attributable Language Models with Retrieval

    Akari Asai, Zexuan Zhong, Danqi Chen, Pang Wei Koh, Luke Zettlemoyer, Hannaneh Hajishirzi, Wen-tau Yih

    arXiv · 2024

    This position paper proposes a roadmap for developing general-purpose retrieval-augmented LMs, which involves a reconsideration of datastores and retrievers, the exploration of pipelines with improved retriever-LM interaction, and significant investment in infrastructure for efficient training and inference.

    99
  • Stronger Data Poisoning Attacks Break Data Sanitization Defenses

    Pang Wei Koh, Jacob Steinhardt, Percy Liang

    arXiv · 2018

    96
  • ReasonIR: Training Retrievers for Reasoning Tasks

    Rulin Shao, Qiao, Rui, Varsha Kishore, Niklas Muennighoff, Xi Lin, Daniela L. Rus, Bryan Kian Hsiang Low, Sewon Min, +3 more

    arXiv · 2025

    ReasonIR-8B is presented, the first retriever specifically trained for general reasoning tasks, and its performance consistently increases with longer and more information-rich rewritten queries; it continues to outperform other retrievers when combined with an LLM reranker.

    88
  • On the Accuracy of Influence Functions for Measuring Group Effects

    Pang Wei Koh, Kai-Siang Ang, Hubert H. K. Teo, Percy Liang

    arXiv · 2019

    82
  • Certified Defenses for Data Poisoning Attacks

    Jacob Steinhardt, Pang Wei Koh, Percy Liang

    arXiv · 2017

    81
  • OpenScholar: Synthesizing Scientific Literature with Retrieval-augmented LMs

    Akari Asai, Jacqueline He, Rulin Shao, Weijia Shi, Amanpreet Singh, Joseph Chee Chang, Kyle Lo, Luca Soldaini, +17 more

    arXiv · 2024

    OpenScholar is introduced, a specialized retrieval-augmented LM that answers scientific queries by identifying relevant passages from 45 million open-access papers and synthesizing citation-backed responses, and achieves citation accuracy on par with human experts.

    80
  • An atlas of transcriptional, chromatin accessibility, and surface marker changes in human mesoderm development

    Pang Wei Koh, Rahul Sinha, Amira A. Barkal, Rachel M. Morganti, Angela C. Chen, Irving L. Weissman, Lay Teng Ang, Anshul Bharat Kundaje, +1 more

    Scientific Data · 2016

    This molecular atlas will facilitate study of human mesoderm development (which cannot be interrogated in vivo due to restrictions on human embryo studies) and provides a broad resource for the study of gene regulation in development at the single-cell level, knowledge that might one day be exploited for regenerative medicine.

    72
  • Just Train Twice: Improving Group Robustness without Training Group Information

    Evan Zheran Liu, Behzad Haghgoo, Annie Chen, Aditi Raghunathan, Pang Wei Koh, Shiori Sagawa, Percy Liang, Chelsea Finn

    arXiv · 2021

    68
  • Denoising genome-wide histone ChIP-seq with convolutional neural networks

    Pang Wei Koh, Emma Pierson, Anshul Bharat Kundaje

    Bioinformatics · 2017

    Aconvolutional denoising algorithm, Coda, is introduced that uses convolutional neural networks to learn a mapping from suboptimal to high-quality histone ChIP-seq data, and has the potential to improve data quality at reduced costs.

    65
  • Supporting COVID-19 Policy Response with Large-scale Mobility-based Modeling

    Serina Y. Chang, Mandy Wilson, Bryan Leroy Lewis, Zakaria Mehrab, Komal K. Dudakiya, Emma Pierson, Pang Wei Koh, Jaline Gerardin, +4 more

    Proceedings of the 27th ACM SIGKDD Conference on Knowledge Discovery & Data Mining · 2021

    This model captures the spread of COVID-19 by using a fine-grained, dynamic mobility network that encodes the hourly movements of people from neighborhoods to individual places, with over 3 billion hourly edges.

    49
  • Uncertainty of Thoughts: Uncertainty-Aware Planning Enhances Information Seeking in Large Language Models

    Zhiyuan Hu, Chumin Liu, Xidong Feng, Yilun Zhao, See Kiong Ng, Anh Tuan Luu, Junxian He, Pang Wei Koh, +1 more

    arXiv · 2024

    Uncertainty of Thoughts is introduced, an algorithm to augment large language models with the ability to actively seek information by asking effective questions and achieves an average performance improvement of 38.1% in the rate of successful task completion across multiple LLMs compared with direct prompting.

    46
  • ExpBERT: Representation Engineering with Natural Language Explanations

    Shikhar Murty, Pang Wei Koh, Percy Liang

    Annual Meeting of the Association for Computational Linguistics (ACL) · 2020

    Across three relation extraction tasks, the method, ExpBERT, matches a BERT baseline but with 3–20x less labeled data and improves on the baseline by 3–10 F1 points with the same amount of labeled data.

    46
  • DataDecide: How to Predict Best Pretraining Data with Small Experiments

    Ian Magnusson, Tai, Nguyen, Ben Bogin, David Heineman, Jena D. Hwang, Luca Soldaini, Akshita Bhagia, Jiacheng Liu, +5 more

    arXiv · 2025

    This work conducts controlled pretraining experiments across 25 corpora with differing sources, deduplication, and filtering up to 100B tokens, model sizes up to 1B parameters, and 3 random seeds and finds that the ranking of models at a single, small size is a strong baseline for predicting best models at the larger target scale (1B).

    43
  • Large-Scale Data Selection for Instruction Tuning

    Hamish Ivison, Zhang, Muru, Faeze Brahman, Pang Wei Koh, Pradeep Dasigi

    arXiv · 2025

    A variant of representation-based data selection (RDS+), which uses weighted mean pooling of pretrained LM hidden states, consistently outperforms more complex methods across all settings tested -- all whilst being more compute-efficient.

    43
  • Dissecting an Online Intervention for Cancer Survivors

    Zhenghao Chen, Pang Wei Koh, Philip Lloyd Ritter, Kate R. Lorig, Erin O’Carroll Bantum, Suchi Saria

    Health Education & Behavior · 2014

    The nature of participant engagement with an online, 6-week workshop for cancer survivors and how such engagement may affect health outcomes is explored and the methods used can be used in analyzing other large electronically mediated social-networking interventions.

    42
  • RLVE: Scaling Up Reinforcement Learning for Language Models with Adaptive Verifiable Environments

    Zhiyuan Zeng, Hamish Ivison, Wang, Yiping, Lifan Yuan, Shuyue Stella Li, Zhuorui Ye, Siting Li, Jacqueline He, +9 more

    arXiv · 2025

    This work introduces Reinforcement Learning with Adaptive Verifiable Environments (RLVE), an approach using verifiable environments that procedurally generate problems and provide algorithmically verifiable rewards, to scale up RL for language models (LMs) and shows that environment scaling consistently improves generalizable reasoning capabilities.

    39
  • Establishing Task Scaling Laws via Compute-Efficient Model Ladders

    Akshita Bhagia, Jiacheng Liu, Alexander Wettig, David Heineman, Oyvind Tafjord, Ananya Harsh Jha, Luca Soldaini, Noah A. Smith, +4 more

    arXiv · 2024

    38
  • ICONS: Influence Consensus for Vision-Language Data Selection

    Xindi Wu, Mengzhou Xia, Rulin Shao, Zhiwei Deng, Pang Wei Koh, Olga Russakovsky

    arXiv · 2024

    This work introduces ICONS, a gradient-based Influence CONsensus approach for vision-language data Selection, which leverages first-order training dynamics to estimate each example's influence on validation performance, then aggregates these estimates across tasks via majority voting.

    35
  • Out-of-Domain Robustness via Targeted Augmentations

    Irena Gao, Shiori Sagawa, Pang Wei Koh, Tatsunori Hashimoto, Percy Liang

    arXiv · 2023

    It is proved that targeted augmentations improve OOD performance, allowing models to generalize better with fewer domains, and set new states-of-the-art for OOD performance by 3.2-15.2 percentage points.

    35
  • Fluid Language Model Benchmarking

    Valentin Hofmann, David Heineman, Ian Magnusson, Kyle Lo, Jesse Dodge, Maarten Sap, Pang Wei Koh, Wang Chun, +2 more

    arXiv · 2025

    Fluid Benchmarking is introduced, a new evaluation approach that advances language model benchmarking across multiple dimensions based on the insight that the relative value of benchmark items depends on an LM's capability level, suggesting that evaluation should adapt to each LM.

    32
  • The Delta Learning Hypothesis: Preference Tuning on Weak Data can Yield Strong Gains

    Scott Geng, Hamish Ivison, Chunliang Li, Maarten Sap, Jerry Li, Ranjay Krishna, Pang Wei Koh

    arXiv · 2025

    This work shows that models can learn surprisingly well from paired data that might typically be considered weak, and enables simpler and cheaper open recipes for state-of-the-art post-training.

    31
  • A False Sense of Privacy: Evaluating Textual Data Sanitization Beyond Surface-level Privacy Leakage

    Rui Xin, Niloofar Mireshghallah, Shuyue Stella Li, Michael Duan, Hyunwoo Kim, Yejin Choi, Yulia Tsvetkov, Sewoong Oh, +1 more

    IEEE Conference on Secure and Trustworthy Machine Learning (SaTML) · 2026

    This work proposes a new framework that evaluates re-identification attacks to quantify individual privacy risks upon data release and shows that seemingly innocuous auxiliary information—such as routine social activities—can be used to infer sensitive attributes like age or substance use history from sanitized data.

    30
  • Proximity-Informed Calibration for Deep Neural Networks

    Miao Xiong, Ailin Deng, Pang Wei Koh, Jiaying Wu, Li Shen, Jianqing Xu, Bryan Hooi

    arXiv · 2023

    ProCal is proposed, a plug-and-play algorithm with a theoretical guarantee to adjust sample confidence based on proximity bias that is effective in addressing proximity bias and improving calibration on balanced, long-tail, and distribution-shift settings under four metrics over various model architectures.

    30
  • Accuracy on the Line: On the Strong Correlation Between Out-of-Distribution and In-Distribution Generalization

    John J. Miller, Rohan Taori, Aditi Raghunathan, Shiori Sagawa, Pang Wei Koh, Vaishaal Shankar, Percy Liang, Yair Carmon, +1 more

    arXiv · 2021

    29
  • Instructional Fingerprinting of Large Language Models

    Jiashu Xu, Fei Wang, Mingyu Ma, Pang Wei Koh, Chaowei Xiao, Muhao Chen

    Conference of the North American Chapter of the Association for Computational Linguistics: Human Language Technologies (NAACL-HLT) · 2024

    23
  • PLeaS — Merging Models with Permutations and Least Squares

    Anshul Nasery, Jonathan Hayase, Pang Wei Koh, Sewoong Oh

    IEEE Computer Society Conference on Computer Vision and Pattern Recognition/Proceedings - IEEE Computer Society Conference on Computer Vision and Pattern Recognition/Proceedings · 2025

    A new two-step algorithm to merge models—termed PLeaS—which allows a practitioner to merge two models sharing the same architecture into a single performant model of a desired size, even when the two original models are fine-tuned from different base models.

    20
  • Temporal FiLM: Capturing Long-Range Sequence Dependencies with Feature-Wise Modulations

    Sawyer Birnbaum, Volodymyr Kuleshov, Zayd Enam, Pang Wei Koh, Stefano Ermon

    arXiv · 2019

    20
  • Using Large Language Models to Promote Health Equity

    Emma Pierson, Divya Shanmugam, Rajiv Movva, Jon M. Kleinberg, Monica Agrawal, Mark H. Dredze, Kadija Ferryman, Judy Wawira Gichoya, +7 more

    NEJM AI · 2025

    17
  • On the Trade-off of Intra-/Inter-class Diversity for Supervised Pre-training

    Jieyu Zhang, Bohan Wang, Hu, Zhengyu, Pang Wei Koh, Alexander Ratner

    arXiv · 2023

    The theory reveals that the optimal class-to-sample ratio (#classes / #samples per class) is invariant to the size of the pre-training dataset, which motivates an application of predicting the optimal number of pre- training classes.

    15
  • Exploring How Generative MLLMs Perceive More Than CLIP with the Same Vision Encoder

    Siting Li, Pang Wei Koh, Simon S. Du

    Annual Meeting of the Association for Computational Linguistics (ACL) · 2025

    This study shows that another branch of Vision-Language Models, Generative Multimodal Large Language Models (MLLMs), achieve significantly higher accuracy than CLIP in many of these tasks using the same vision encoder and weights, indicating that these Generative MLLMs perceive more -- as they extract and utilize visual information more effectively.

    13
  • EvoLM: Self-Evolving Language Models through Co-Evolved Discriminative Rubrics

    Shuyue Stella Li, Rui Xin, Teng Xiao, Yike Wang, Rulin Shao, Zoey Hao, Melanie Sclar, Sewoong Oh, +3 more

    arXiv · 2026

    EVOLM demonstrates that structuring a model's evaluative capacity into co-evolving discriminative rubrics enables self-improvement without external supervision.

    12
  • Extending the WILDS Benchmark for Unsupervised Adaptation

    Shiori Sagawa, Pang Wei Koh, Tong Lee, Irena Gao, Sang Michael Xie, Kendrick Shen, Ananya Kumar, Weihua Hu, +12 more

    arXiv · 2021

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

    Jeffrey Ji-Peng Li, Alex Fang, Georgios Smyrnis, Maor Ivgi, Matt Jordan, Samir Yitzhak Gadre, Hritik Bansal, Etash Guha, +22 more

    neural information processing systems · 2024

    10
  • Localized hepatic lobular regeneration by central-vein–associated lineage-restricted progenitors

    Jonathan M. Tsai, Pang Wei Koh, Ania Stefańska, Liujing Xing, Graham G. Walmsley, Nicolas Poux, Irving L. Weissman, Yuval Rinkevich

    Proceedings of the National Academy of Sciences · 2017

    A previously unknown program of liver regeneration after acute injury is discovered in the neonatal mouse liver, specific to the first week of life, where numerous rounds of cell division and reconstitution of lobe architecture much like in amphibian limbs are observed.

    10
  • Annotation alignment: Comparing LLM and human annotations of conversational safety

    Rajiv Movva, Pang Wei Koh, Emma Pierson

    Conference on Empirical Methods in Natural Language Processing (EMNLP) · 2024

    9
  • Information-Theoretic Distillation for Reference-less Summarization

    Jaehun Jung, Ximing Lu, Liwei Jiang, Faeze Brahman, West, Peter, Pang Wei Koh, Yejin Choi

    arXiv · 2024

    9
  • Frustratingly Simple Retrieval Improves Challenging, Reasoning-Intensive Benchmarks

    Xinxi Lyu, Michael Duan, Rulin Shao, Pang Wei Koh, Sewon Min

    arXiv · 2025

    CompactDS is introduced: a diverse, high-quality, web-scale datastore that achieves high retrieval accuracy and subsecond latency on a single-node, and its carefully designed in-house datastore matches or outperforms web search engines such as Google Search, as well as recently proposed, complex agent-based RAG systems--all while maintaining simplicity, reproducibility, and self-containment.

    8
  • Scaling Retrieval-Based Language Models with a Trillion-Token Datastore

    Rulin Shao, Jacqueline He, Akari Asai, Weijia Shi, Tim Dettmers, Sewon Min, Luke Zettlemoyer, Pang Wei Koh

    neural information processing systems · 2024

    8
  • Inferring Multidimensional Rates of Aging from Cross-Sectional Data

    Emma Pierson, Pang Wei Koh, Tatsunori Hashimoto, Daphne Koller, Jure Leskovec, Nicholas Eriksson, Percy Liang

    PubMed · 2018

    8
  • JobBench: Aligning Agent Work With Human Will

    Yuetai Li, Yichen Feng, Zhangchen Xu, Zixian Ma, Kaiyuan Zheng, Fengqing Jiang, Xinghua Sun, Rulin Shao, +16 more

    arXiv · 2026

    JobBench is introduced, which evaluates AI agents on the workflows that experts identify as high-priority for delegation, empowering humans based on their needs instead of replacing them with GDP value, and hopes it shifts the community's target labour-market effect from replacement to enhancement.

    7
  • The Unmet Promise of Synthetic Training Images: Using Retrieved Real Images Performs Better

    Scott Geng, Cheng-Yu Hsieh, Vivek Ramanujan, Matthew Wallingford, Chunliang Li, Pang Wei Koh, Ranjay Krishna

    neural information processing systems · 2024

    7
  • Self-Improving VLM Judges Without Human Annotations

    Inna Wanyin Lin, Yushi Hu, Shuyue Stella Li, Scott Geng, Pang Wei Koh, Luke Zettlemoyer, Tim Althoff, Marjan Ghazvininejad

    arXiv · 2025

    This work presents a framework to self-train a VLM judge model without any human preference annotations, using only self-synthesized data, and shows strong gains in general, hallucination, and reasoning dimensions.

    6
  • OLMoTrace: Tracing Language Model Outputs Back to Trillions of Training Tokens

    Jiacheng Liu, Taylor Blanton, Yanai Elazar, Sewon Min, Yen-Sung Chen, Arnavi Chheda-Kothary, Huy Tran, Byron Bischoff, +22 more

    Annual Meeting of the Association for Computational Linguistics (ACL) · 2025

    6
  • JPEG-LM: LLMs as Image Generators with Canonical Codec Representations

    Xiaochuang Han, Marjan Ghazvininejad, Pang Wei Koh, Yulia Tsvetkov

    arXiv · 2024

    It is shown that using canonical codec representations can help lower the barriers between language generation and visual generation, facilitating future research on multi-modal language/image/video LLMs.

    6
  • Olmo 3

    Team OLMo, :, Allyson Ettinger, Amanda Bertsch, Bailey Kuehl, Graham, David, David Heineman, Dirk Groeneveld, +22 more

    arXiv · 2025

    5
  • Metabolically purified human stem cell-derived hepatocytes reveal distinct effects of Ebola and Lassa viruses

    Joseph B. Prescott, Kevin J. Liu, Angelika Lander, Nicole Min Qian Pek, Sawan Kumar Jha, Marcel Bokelmann, Manali Begur, Pang Wei Koh, +6 more

    bioRxiv (Cold Spring Harbor Laboratory) · 2025

    A new metabolic selection approach is introduced to create nearly-pure hepatocytes from human pluripotent stem cells, killing non-liver cells by withholding essential nutrients and presenting a single-cell transcriptional and chromatin accessibility roadmap of human hepatocyte differentiation, purification, and viral infection.

    5
  • Negative Token Merging: Image-based Adversarial Feature Guidance

    Jaskirat Singh, Lindsey Li, Weijia Shi, Ranjay Krishna, Yejin Choi, Pang Wei Koh, Michael F. Cohen, Stephen Jay Gould, +2 more

    arXiv · 2024

    5
  • Data-Centric AI in the Age of Large Language Models

    Xinyi Xu, Zhaoxuan Wu, Rui Qiao, Arun Verma, Shu, Yao, Jingtan Wang, Xinyuan Niu, Zhenfeng He, +11 more

    arXiv · 2024

    A data-centric viewpoint of AI research, focusing on large language models (LLMs), is proposed, covering data-centric benchmarks and data curation, data attribution, knowledge transfer, knowledge transfer, and inference contextualization.

    5
  • Selective Classification Can Magnify Disparities Across Groups

    Erik Jones, Shiori Sagawa, Pang Wei Koh, Ananya Kumar, Percy Liang

    arXiv · 2020

    5
  • Privasis: Synthesizing the Largest "Public" Private Dataset from Scratch

    Hyunwoo Kim, Niloofar Mireshghallah, Michael Duan, Rui Xin, Shuyue Stella Li, Jaehun Jung, David Acuna, Qi Pang, +6 more

    arXiv · 2026

    Privasis is presented, the first million-scale fully synthetic dataset entirely built from scratch--an expansive reservoir of texts with rich and diverse private information--designed to broaden and accelerate research in areas where processing sensitive social data is inevitable.

    4
  • Using unlabeled data to enhance fairness of medical AI

    Rajiv Movva, Pang Wei Koh, Emma Pierson

    Nature Medicine · 2024

    4
  • CopyBench: Measuring Literal and Non-Literal Reproduction of Copyright-Protected Text in Language Model Generation

    Tong Chen, Akari Asai, Niloofar Mireshghallah, Sewon Min, James Grimmelmann, Yejin Choi, Hannaneh Hajishirzi, Luke Zettlemoyer, +1 more

    Conference on Empirical Methods in Natural Language Processing (EMNLP) · 2024

    4
  • Improving Domain Generalization with Domain Relations

    Huaxiu Yao, Xinyu Yang, Xinyi Pan, Shengchao Liu, Pang Wei Koh, Chelsea Finn

    arXiv · 2023

    This paper focuses on domain shifts, which occur when the model is applied to new domains that are different from the ones it was trained on, and proposes a new approach called D 3 G, which learns domain-specific models by leveraging the relations among different domains by leveraging the relations among different domains.

    4
  • Deep Reasoning in General Purpose Agents via Structured Meta-Cognition

    Dean Light, Michael Theologitis, Kshitish Ghate, Shuyue Stella Li, Benjamin Newman, Chirag Shah, Aylin Caliskan, Pang Wei Koh, +2 more

    arXiv · 2026

    DOLORES outperforms all evaluated scaffolds across three model sizes and two model families, improving over the strongest evaluated scaffold baseline by 24.8% on average, and points toward future agentic systems that treat scaffolding as adaptive reasoning, constructing the structure each task requires just-in-time.

    2
  • Position Paper: Data-Centric AI in the Age of Large Language Models

    Xinyi Xu, Zhaoxuan Wu, Rui Qiao, Arun Verma, Yao Shu, Jingtan Wang, Xinyuan Niu, Zhenfeng He, +11 more

    Findings of the Association for Computational Linguistics: EMNLP 2024 · 2024

    2
  • Lower bound on the time complexity of local adiabatic evolution

    Zhenghao Chen, Pang Wei Koh, Yan Zhao

    Physical Review A · 2006

    Two different approaches of evaluating the time complexity for local adiabatic evolution using time-independent parameters are presented, thus providing effective tests (not requiring the evaluation of the entire time-dependent gap function) for theTime complexity of newly developed algorithms.

    2
  • Reliable and Responsible Foundation Models: A Comprehensive Survey

    Xinyu Yang, Junlin Han, Rishi Bommasani, Jinqi Luo, Wenjie Qu, Wangchunshu Zhou, Adel Bibi, Xiyao Wang, +22 more

    arXiv · 2026

    Critical issues, including bias and fairness, security and privacy, uncertainty, explainability, and distribution shift are explored, as well as methods like alignment and Artificial Intelligence-Generated Content (AIGC) detection.

    1
  • FlexOLMo: Open Language Models for Flexible Data Use

    Weijia Shi, Akshita Bhagia, Kevin Farhat, Niklas Muennighoff, Jacob Morrison, Evan Pete Walsh, Dustin Schwenk, Shayne Longpre, +15 more

    neural information processing systems · 2025

    1
  • Multilingual Diversity Improves Vision-Language Representations

    Thao D. Nguyen, Matthew Wallingford, Sebastin Santy, Wei-Chiu Ma, Sewoong Oh, Ludwig Schmidt, Pang Wei Koh, Ranjay Krishna

    neural information processing systems · 2024

    1
  • Time Series Super Resolution withTemporal Adaptive Batch Normalization

    Volodymyr Kuleshov, Sawyer Birnbaum, Zayd Enam, Pang Wei Koh, Stefano Ermon

    2018

    A deep neural network architecture for reconstructing a high resolution time series signal from low-resolution measurements, a task that is called time series super resolution.

    1
  • Mesoderm ATAC-seq QC figures

    Pang Wei Koh, Kyle M. Loh, Rahul Sinha, Lay Teng Ang, Irving L. Weissman, Anshul Bharat Kundaje

    Figshare · 2016

    1
  • High-throughput surface marker screening data on a mesoderm differentiation dataset

    Kyle M. Loh, Angela Chen, Pang Wei Koh, Rahul Sinha, Lay Teng Ang, Irving L. Weissman

    Figshare · 2016

    1
  • Annotated bulk-population RNA-seq data on a mesoderm differentiation dataset

    Pang Wei Koh, Zhenghao Chen, Kyle M. Loh, Rahul Sinha, Lay Teng Ang, Irving L. Weissman, Anshul Bharat Kundaje

    Figshare · 2016

    1
  • Spokes: Optimizing for Diverse Pretraining Data Selection

    Clarence Lee, Yejin Choi, Luke Zettlemoyer, Pang Wei Koh, Hai Leong Chieu

    arXiv · 2026

    This work directly optimize diversity by introducing a probabilistic diversification framework based on the G-Vendi score, optimized via exponentiated gradient descent, which produces subsets that are substantially more diverse than those obtained via random sampling.

    –
  • FrugalEvo: Towards Cost-Aware LLM-Guided Program Evolution

    Hui Chen, Xuan Qi, James Zhao, Zhaopeng Feng, Shilong Liu, Kuang Xu, Pang Wei Koh, Bryan Hooi

    arXiv · 2026

    –
  • Knowledge Distillation During Mid-Training Favors Reasoning over Factual Recall

    Jacqueline He, Howard Yen, Shuyue Stella Li, Margaret Li, Hanqing Zeng, Yinglong Xia, Benyu Zhang, Zhuokai Zhao, +4 more

    arXiv · 2026

    –
  • Companion data for CytED construction software

    Lukas Oesinghaus, Molly Park, Rulin Shao, Pang Wei Koh, Georg Seelig

    Zenodo (CERN European Organization for Nuclear Research) · 2026

    –
  • Efficient Test-Time Adaptation through Human-AI Interaction

    Zora Zhiruo Wang, Apurva Gandhi, Rulin Shao, Aspen Chen, Jonas Mueller, Zhiqi Liang, Jett Chen, Michael Ryan, +17 more

    arXiv · 2026

    –
  • CytED and test datasets for pyCytED

    Lukas Oesinghaus, Molly Park, Rulin Shao, Pang Wei Koh, Georg Seelig

    Zenodo (CERN European Organization for Nuclear Research) · 2026

    –
  • Rubric-Based Optimization for Text-to-Music Generation

    Ping Wang, Guang Yang, SU Shao-rong, Junkai Wu, Pang Wei Koh, Noah A. Smith

    arXiv · 2026

    –
  • Precise Information Control in Long-Form Text Generation

    Jacqueline He, Howard Yen, Margaret Li, Shuyue Stella Li, Zhiyuan Zeng, Weijia Shi, Yulia Tsvetkov, Danqi Chen, +2 more

    neural information processing systems · 2025

    –
  • Beyond the Stethoscope: Operationalizing Interactive Clinical Reasoning in Large Language Models via Proactive Information Seeking

    Shuyue Stella Li, Vidhisha Balachandran, Shangbin Feng, Emma Pierson, Pang Wei Koh, Yulia Tsvetkov

    IEEE International Conference on Healthcare Informatics (ICHI) · 2024

    –
  • Merge to Learn: Efficiently Adding Skills to Language Models with Model Merging

    Jacob Morrison, Noah A. Smith, Hannaneh Hajishirzi, Pang Wei Koh, Jesse Dodge, Pradeep Dasigi

    Findings of the Association for Computational Linguistics: EMNLP 2024 · 2024

    –
  • Overparameterization hurts worst-group accuracy with spurious correlations

    Shiori Sagawa, Aditi Raghunathan, Pang Wei Koh, Percy Liang

    International Conference on Machine Learning · 2020

    –
  • Identifying and exploiting influential training examples.

    Pang Wei Koh

    National Conference on Artificial Intelligence · 2019

    –

Publication data from OpenAlex, with missing venues and authors filled in from Crossref; citation counts are the higher of OpenAlex and Semantic Scholar, last synced 2026-10-12. One-sentence summaries under some papers are written by Semantic Scholar’s model. Citation counts may be lower than on Google Scholar, which indexes more sources.

Report an error

Wrong papers, two people merged into one, or a profile that should not be here? Tell us and we will fix or hide it.

Sign in to report