← Search

Dawn Song

134 accepted papers

2026

AgentSynth: Scalable Task Generation for Generalist Computer-Use Agents

ICLR 2026poster

We introduce AgentSynth, a scalable and cost-efficient pipeline for automatically synthesizing high-quality tasks and trajectory datasets for generalist computer-use agents. Leveraging information asymmetry, AgentSynth constructs subtasks that are simple during generation but significantly more chal…

Cited by 0SourcecodeScholar
2026

Can Small Training Runs Reliably Guide Data Curation? Rethinking Proxy-Model Practice

ICLR 2026poster

Data teams at frontier AI companies routinely train small proxy models to make critical decisions about pretraining data recipes for full-scale training. However, the community has a limited understanding of whether and when conclusions drawn from small-scale experiments reliably transfer to large-s…

Cited by 0SourcecodeScholar
2026

Characterizing Agents in Production

ICML 2026oral

LLM-based agents already operate in production across many industries, yet we lack a clear understanding of which technical methods make these deployments successful. We present the first systematic study of Characterizing Agents in Production (CAP) using first-hand data from agent developers. We co…

Cited by 0SourceScholar
2026

CyberCycle: Scalable Real-World Benchmark for AI Agents' End-to-End Cybersecurity Capabilities

ICML 2026poster

AI has the potential to transform cybersecurity by enabling systems that can autonomously detect, analyze, and remediate software vulnerabilities. However, existing cybersecurity evaluations of AI systems are limited in scale or scope, and fail to capture the end-to-end lifecycle of real-world softw…

Cited by 0SourceScholar
2026

CyberGym: Evaluating AI Agents' Real-World Cybersecurity Capabilities at Scale

ICLR 2026oral

AI agents have significant potential to reshape cybersecurity, making a thorough assessment of their capabilities critical. However, existing evaluations fall short, because they are based on small-scale benchmarks and only measure static outcomes, failing to capture the full, dynamic range of real-…

Cited by 0SourceScholar
2026

DELTA-Code: How RL Unlocks and Transfers New Programming Algorithms in LLMs

ICLR 2026poster

It remains an open question whether LLMs can acquire or generalize genuinely new reasoning strategies, beyond the sharpened skills encoded in their parameters during pre-training or post-training. To attempt to answer this debate, we introduce DELTA-Code —Distributional Evaluation of Learnability an…

Cited by 0SourcecodeScholar
2026

DevOps-Gym: Benchmarking AI Agents in Software DevOps Cycle

ICLR 2026poster

Even though demonstrating extraordinary capabilities in code generation and software issue resolving, AI agents' capabilities in the full software DevOps cycle are still unknown. Different from pure code generation, handling the DevOps cycle in real-world software, including developing, deploying, a…

Cited by 0SourcecodeScholar
2026

FrontierCS: Evolving Challenges for Evolving Intelligence

ICML 2026poster

We introduce FrontierCS, a benchmark of 240 open-ended problems across diverse areas of computer science, designed and reviewed by experts, including CS PhDs and top-tier competitive programming participants and problem setters. Unlike existing benchmarks that focus on tasks with known optimal solut…

Cited by 0SourceScholar
2026

Holistic Agent Leaderboard: The Missing Infrastructure for AI Agent Evaluation

ICLR 2026poster

AI agents have been developed for complex real-world tasks from coding to customer service. But AI agent evaluations suffer from many challenges that undermine our understanding of how well agents really work (Figure 1). We introduce the Holistic Agent Leaderboard (HAL) to address these challenges.…

Cited by 0SourcecodeScholar
2026

In-Context Watermarks for Large Language Models

ICLR 2026poster

The growing use of large language models (LLMs) for sensitive applications has highlighted the need for effective watermarking techniques to ensure the provenance and accountability of AI-generated text. However, most existing watermarking methods require access to the decoding process, limiting the…

Cited by 0SourcecodeScholar
2026

OpenSage: Self-programming Agent Generation Engine

ICML 2026poster

Agent development kits (ADKs) provide effective platforms and tooling for constructing agents, and their designs are critical to the constructed agents' performance, especially the functionality for agent topology, tools, and memory. However, current ADKs either lack sufficient functional support or…

Cited by 0SourceScholar
2026

Position: Agent Evaluation Should Be Agentified for Openness, Standardization, and Reproducibility

ICML 2026poster

Agent systems are advancing quickly across domains, but their evaluation remains fragmented. Most benchmarks rely on fixed, LLM-centric harnesses that require heavy integration, create test-production mismatch, and limit fair comparison across diverse agent designs. This position paper argues that t…

Cited by 0SourceScholar
2026

Position: Agent Security Needs Redefinition through a Holistic Framework

ICML 2026poster

Existing definitions of agent security are ambiguous because they do not fully capture the holistic view across agent components. For instance, current work fails to distinguish between potentially legitimate administrative tasks and malicious exploitation of the same command. A command to "delete u…

Cited by 0SourceScholar
2026

RedCodeAgent: Automatic Red-teaming Agent against Diverse Code Agents

ICLR 2026poster

Code agents have gained widespread adoption due to their strong code generation capabilities and integration with code interpreters, enabling dynamic execution, debugging, and interactive programming capabilities. While these advancements have streamlined complex workflows, they have also introduced…

Cited by 0SourcecodeScholar
2026

RepIt: Steering Language Models with Concept-Specific Refusal Vectors

ICLR 2026poster

Current safety evaluations of language models rely on benchmark-based assessments that may miss targeted vulnerabilities. We present RepIt, a simple and data-efficient framework for isolating concept-specific representations in LM activations. While existing steering methods already achieve high att…

Cited by 0SourcecodeScholar
2026

Representational Similarity and Model Behavior in Multi-Agent Interaction

ICML 2026poster

Researchers have shown that neural similarity among humans predicts social closeness and cooperative success, whereas innovation often emerges from interactions among dissimilar individuals. We investigate whether these principles extend to artificial intelligence by examining interactions between l…

Cited by 0SourceScholar
2026

Signal in the Noise: Polysemantic Interference Transfers and Predicts Cross-Model Influence

ICLR 2026poster

Polysemanticity is pervasive in language models and remains a major challenge for interpretation and model behavioral control. Leveraging sparse autoencoders (SAEs), we map the polysemantic topology of two small models (Pythia-70M and GPT-2-Small) to identify SAE feature pairs that are semantically…

Cited by 0SourceScholar
2026

SteeringSafety: Benchmarking Representation Steering in LLMs Across Safety Perspectives

ICML 2026poster

We introduce STEERINGSAFETY, a benchmark for evaluating representation steering methods across nine safety perspectives spanning 18 datasets. While prior work highlights general capabilities of representation steering, we focus on safety perspectives including bias, harmfulness, hallucination, socia…

Cited by 0SourceScholar
2026

Strategy Executability in Mathematical Reasoning: Leveraging Human–Model Differences for Effective Guidance

ICML 2026poster

Example-based guidance is widely used to improve mathematical reasoning at inference time, yet its effectiveness is highly unstable across problems and models—even when the guidance is correct and problem-relevant. We show that this instability arises from a previously underexplored gap between *str…

Cited by 0SourceScholar
2026

TrustGen: A Platform of Dynamic Benchmarking on the Trustworthiness of Generative Foundation Models

ICLR 2026poster

Generative foundation models (GenFMs), such as large language models and text-to-image systems, have demonstrated remarkable capabilities in various downstream applications. As they are increasingly deployed in high-stakes applications, assessing their trustworthiness has become both a critical nece…

Cited by 0SourceScholar
2026

Unsafer in Many Turns: Benchmarking and Defending Multi-Turn Safety Risks in Tool-Using Agents

ICML 2026poster

LLM-based agents are becoming increasingly capable, yet their safety lags behind. This creates a gap between what agents can do and should do. This gap widens as agents engage in multi-turn interactions and employ diverse tools, introducing new risks overlooked by existing benchmarks. To systematica…

Cited by 0SourceScholar
2026

VERINA: Benchmarking Verifiable Code Generation

ICLR 2026poster

Large language models (LLMs) are increasingly integrated in software development, but ensuring correctness in LLM-generated code remains challenging and often requires costly manual review. Verifiable code generation---jointly generating code, specifications, and proofs of code-specification alignme…

Cited by 0SourcecodeScholar
2026

When Benign Inputs Lead to Severe Harms: Eliciting Unsafe Unintended Behaviors of Computer-Use Agents

ICML 2026poster

Although computer-use agents (CUAs) hold significant potential to automate increasingly complex OS workflows, they can demonstrate unsafe unintended behaviors that deviate from expected outcomes even under benign input contexts. However, exploration of this risk remains largely anecdotal, lacking co…

Cited by 0SourceScholar
2025

A Sustainable AI Economy Needs Data Deals That Work for Generators

NeurIPS 2025poster

We argue that the machine learning value chain is structurally unsustainable due to an economic data processing inequality: each state in the data cycle from inputs to model weights to synthetic outputs refines technical signal but strips economic equity from data generators. We show, by analyzing s…

Cited by 0SourceScholar
2025

AGENTVIGIL: Automatic Black-Box Red-teaming for Indirect Prompt Injection against LLM Agents

EMNLP 2025

There emerges a critical security risk of LLM agents: indirect prompt injection, a sophisticated attack vector that compromises thecore of these agents, the LLM, by manipulating contextual information rather than direct user prompts. In this work, we propose a generic black-box optimization framewor

Cited by 0SourcePDFScholar
2025

AIR-BENCH 2024: A Safety Benchmark based on Regulation and Policies Specified Risk Categories

ICLR 2025spotlight

Foundation models (FMs) provide societal benefits but also amplify risks. Governments, companies, and researchers have proposed regulatory frameworks, acceptable use policies, and safety benchmarks in response. However, existing public benchmarks often define safety categories based on previous lite…

Cited by 0SourcePDFScholar
2025

BountyBench: Dollar Impact of AI Agent Attackers and Defenders on Real-World Cybersecurity Systems

NeurIPS 2025poster

AI agents have the potential to significantly alter the cybersecurity landscape. Here, we introduce the first framework to capture offensive and defensive cyber-capabilities in evolving real-world systems. Instantiating this framework with BountyBench, we set up 25 systems with complex, real-world c…

Cited by 0SourceScholar
2025

COSMIC: Generalized Refusal Direction Identification in LLM Activations

ACL 2025finding

Large Language Models encode behaviors like refusal within their activation space, but identifying these behaviors remains challenging. Existing methods depend on predefined refusal templates detectable in output tokens or manual review. We introduce **COSMIC** (Cosine Similarity Metrics for Inversi…

2025

Capturing the Temporal Dependence of Training Data Influence

ICLR 2025oral

Traditional data influence estimation methods, like influence function, assume that learning algorithms are permutation-invariant with respect to training data. However, modern training paradigms—especially for foundation models using stochastic algorithms and non-convergent, multi-stage curricula—a…

Cited by 2SourcePDFScholar
2025

CodeHalu: Investigating Code Hallucinations in LLMs via Execution-based Verification

AAAI 2025technical

Large Language Models (LLMs) have made significant progress in code generation, offering developers groundbreaking automated programming support. However, LLMs often generate code that is syntactically correct and even semantically plausible, but may not execute as expected or fulfill specified requ…

2025

GuardAgent: Safeguard LLM Agents via Knowledge-Enabled Reasoning

ICML 2025poster

The rapid advancement of large language model (LLM) agents has raised new concerns regarding their safety and security. In this paper, we propose GuardAgent, the first guardrail agent to protect target agents by dynamically checking whether their actions satisfy given safety guard requests. Specific…

2025

Improving LLM Safety Alignment with Dual-Objective Optimization

ICML 2025poster

Existing training-time safety alignment techniques for large language models (LLMs) remain vulnerable to jailbreak attacks. Direct preference optimization (DPO), a widely deployed alignment method, exhibits limitations in both experimental and theoretical contexts as its loss function proves subopti…

2025

MMDT: Decoding the Trustworthiness and Safety of Multimodal Foundation Models

ICLR 2025poster

Multimodal foundation models (MMFMs) play a crucial role in various applications, including autonomous driving, healthcare, and virtual assistants. However, several studies have revealed vulnerabilities in these models, such as generating unsafe content by text-to-image models. Existing benchmarks o…

2025

Multimodal Situational Safety

ICLR 2025poster

Multimodal Large Language Models (MLLMs) are rapidly evolving, demonstrating impressive capabilities as multimodal assistants that interact with both humans and their environments. However, this increased sophistication introduces significant safety concerns. In this paper, we present the first eval…

Cited by 5SourcePDFScholar
2025

OMEGA: Can LLMs Reason Outside the Box in Math? Evaluating Exploratory, Compositional, and Transformative Generalization

NeurIPS 2025poster

Recent large language models (LLMs) with long-chain-of-thought reasoning—such as DeepSeek-R1—have achieved impressive results on Olympiad-level mathematics benchmarks. However, they often rely on a narrow set of strategies and struggle with problems that require a novel way of thinking. To systemati…

Cited by 0SourcecodeScholar
2025

OVERT: A Benchmark for Over-Refusal Evaluation on Text-to-Image Models

NeurIPS 2025poster

Text-to-Image (T2I) models have achieved remarkable success in generating visual content from text inputs. Although multiple safety alignment strategies have been proposed to prevent harmful outputs, they often lead to overly cautious behavior ---rejecting even benign prompts---a phenomenon known as…

Cited by 0SourcecodeScholar
2025

Position: Formal Mathematical Reasoning—A New Frontier in AI

ICML 2025spotlight

AI for Mathematics (AI4Math) is intellectually intriguing and is crucial for AI-driven system design and verification. Extensive efforts on AI4Math have mirrored techniques in NLP, in particular, training large language models on carefully curated math datasets in text form. As a complementary yet l…

Cited by 0SourcePDFScholar
2025

Position: In-House Evaluation Is Not Enough. Towards Robust Third-Party Evaluation and Flaw Disclosure for General-Purpose AI

ICML 2025spotlight

The widespread deployment of general-purpose AI (GPAI) systems introduces significant new risks. Yet the infrastructure, practices, and norms for reporting flaws in GPAI systems remain seriously underdeveloped, lagging far behind more established fields like software security. Based on a collaborati…

Cited by 0SourcePDFScholar
2025

Position: Political Neutrality in AI Is Impossible — But Here Is How to Approximate It

ICML 2025oral

AI systems often exhibit political bias, influencing users' opinions and decisions. While political neutrality—defined as the absence of bias—is often seen as an ideal solution for fairness and safety, this position paper argues that true political neutrality is neither feasible nor universally desi…

Cited by 0SourcePDFScholar
2025

SECODEPLT: A Unified Benchmark for Evaluating the Security Risks and Capabilities of Code GenAI

NeurIPS 2025poster

Existing benchmarks for evaluating the security risks and capabilities (e.g., vulnerability detection) of code-generating large language models (LLMs) face several key limitations: (1) limited coverage of risk and capabilities; (2) reliance on static evaluation metrics such as LLM judgments or rule-…

Cited by 0SourcecodeScholar
2025

SafeKey: Amplifying Aha-Moment Insights for Safety Reasoning

EMNLP 2025

Large Reasoning Models (LRMs) introduce a new generation paradigm of explicitly reasoning before answering, leading to remarkable improvements in complex tasks. However, they pose great safety risks against harmful queries and adversarial attacks. While recent mainstream safety efforts on LRMs, supe

2025

Scalable Best-of-N Selection for Large Language Models via Self-Certainty

NeurIPS 2025poster

Best-of-N selection is a key technique for improving the reasoning performance of Large Language Models (LLMs) through increased test-time computation. Current state-of-the-art methods often employ computationally intensive reward models for response evaluation and selection. Reward-free alternative…

Cited by 0SourcecodeScholar
2025

Tamper-Resistant Safeguards for Open-Weight LLMs

ICLR 2025poster

Rapid advances in the capabilities of large language models (LLMs) have raised widespread concerns regarding their potential for malicious use. Open-weight LLMs present unique challenges, as existing safeguards lack robustness to tampering attacks that modify model weights. For example, recent works…

2025

VMDT: Decoding the Trustworthiness of Video Foundation Models

NeurIPS 2025poster

As foundation models become more sophisticated, ensuring their trustworthiness becomes increasingly critical; yet, unlike text and image, the video modality still lacks comprehensive trustworthiness benchmarks. We introduce VMDT (Video-Modal DecodingTrust), the first unified platform for evaluating…

Cited by 0SourcecodeScholar
2025

Why and How LLMs Hallucinate: Connecting the Dots with Subsequence Associations

NeurIPS 2025poster

Large language models (LLMs) frequently generate hallucinations—content that deviates from factually inaccurate or deviates from provided context—posing challenges for diagnosis. However, diagnosing the causes of hallucination is challenging due to the complex interplay of underlying causes. This pa…

Cited by 0SourcecodeScholar
2024

Agent Instructs Large Language Models to be General Zero-Shot Reasoners

ICML 2024poster

We introduce a method to improve the zero-shot reasoning abilities of large language models on general language understanding tasks. Specifically, we build an autonomous agent to instruct the reasoning process of large language models. To enable this, our agent only needs to generate a single set of…

2024

AgentPoison: Red-teaming LLM Agents via Poisoning Memory or Knowledge Bases

NeurIPS 2024poster

LLM agents have demonstrated remarkable performance across various applications, primarily due to their advanced capabilities in reasoning, utilizing external knowledge and tools, calling APIs, and executing actions to interact with environments. Current agents typically utilize a memory module or a…

2024

BEEAR: Embedding-based Adversarial Removal of Safety Backdoors in Instruction-tuned Language Models

EMNLP 2024main

Safety backdoor attacks in large language models (LLMs) enable harmful behaviors to be stealthily triggered while evading detection during normal interactions. The high dimensionality of the trigger search space and the diverse range of potential malicious behaviors in LLMs make this a critical open…

2024

Boosting Alignment for Post-Unlearning Text-to-Image Generative Models

NeurIPS 2024poster

Large-scale generative models have shown impressive image-generation capabilities, propelled by massive data. However, this often inadvertently leads to the generation of harmful or inappropriate content and raises copyright concerns. Driven by these concerns, machine unlearning has become crucial t…

2024

C-RAG: Certified Generation Risks for Retrieval-Augmented Language Models

ICML 2024poster

Despite the impressive capabilities of large language models (LLMs) across diverse applications, they still suffer from trustworthiness issues, such as hallucinations and misalignments. Retrieval-augmented language models (RAG) have been proposed to enhance the credibility of generations by groundin…

2024

Decoding Compressed Trust: Scrutinizing the Trustworthiness of Efficient LLMs Under Compression

ICML 2024poster

Compressing high-capability Large Language Models (LLMs) has emerged as a favored strategy for resource-efficient inferences. While state-of-the-art (SoTA) compression methods boast impressive advancements in preserving benign task performance, the potential risks of compression in terms of safety a…

2024

Effective and Efficient Federated Tree Learning on Hybrid Data

ICLR 2024poster

Federated learning has emerged as a promising distributed learning paradigm that facilitates collaborative learning among multiple parties without transferring raw data. However, most existing federated learning studies focus on either horizontal or vertical data settings, where the data of differen…

Cited by 5SourcePDFScholar
2024

GREATS: Online Selection of High-Quality Data for LLM Training in Every Iteration

NeurIPS 2024spotlight

Online batch selection methods offer an adaptive alternative to static training data selection by dynamically selecting data batches during training. However, existing methods either rely on impractical reference models or simple heuristics that may not capture true data informativeness. To address…

Cited by 6SourcePDFScholar
2024

Hidden Persuaders: LLMs’ Political Leaning and Their Influence on Voters

EMNLP 2024main

Do LLMs have political leanings and are LLMs able to shift our political views? This paper explores these questions in the context of the 2024 U.S. presidential election. Through a voting simulation, we demonstrate 18 open-weight and closed-source LLMs’ political preference for Biden over Trump. We…

2024

Position: Evolving AI Collectives Enhance Human Diversity and Enable Self-Regulation

ICML 2024poster

Large language model behavior is shaped by the language of those with whom they interact. This capacity and their increasing prevalence online portend that they will intentionally or unintentionally "program" one another and form emergent AI subjectivities, relationships, and collectives. Here, we c…

Cited by 6SourcePDFScholar
2024

Position: On the Societal Impact of Open Foundation Models

ICML 2024oral

Foundation models are powerful technologies: how they are released publicly directly shapes their societal impact. In this position paper, we focus on *open* foundation models, defined here as those with broadly available model weights (e.g., Llama 3, Stable Diffusion XL). We identify five distincti…

Cited by 4SourcePDFScholar
2024

Re-Tuning: Overcoming the Compositionality Limits of Large Language Models with Recursive Tuning

ACL 2024long

We present a new method for large language models to solve compositional tasks. Although they have shown strong performance on traditional language understanding tasks, large language models struggle to solve compositional tasks, where the solution depends on solving smaller instances of the same pr…

2024

RedCode: Risky Code Execution and Generation Benchmark for Code Agents

NeurIPS 2024poster

With the rapidly increasing capabilities and adoption of code agents for AI-assisted coding and software development, safety and security concerns, such as generating or executing malicious code, have become significant barriers to the real-world deployment of these agents. To provide comprehensive…

Cited by 7SourcecodeScholar
2024

RigorLLM: Resilient Guardrails for Large Language Models against Undesired Content

ICML 2024poster

Recent advancements in Large Language Models (LLMs) have showcased remarkable capabilities across various tasks in different domains. However, the emergence of biases and the potential for generating harmful content in LLMs, particularly under malicious inputs, pose significant challenges. Current m…

2024

The False Promise of Imitating Proprietary Language Models

ICLR 2024spotlight

An emerging method to cheaply improve a weaker language model is to finetune it on outputs from a stronger model, such as a proprietary system like ChatGPT (e.g., Alpaca, Self-Instruct, and others). In this work, we critically analyze this approach of imitating language models. We first finetune a s…

Cited by 11SourcePDFScholar
2023

BIRD: Generalizable Backdoor Detection and Removal for Deep Reinforcement Learning

NeurIPS 2023poster

Backdoor attacks pose a severe threat to the supply chain management of deep reinforcement learning (DRL) policies. Despite initial defenses proposed in recent studies, these methods have very limited generalizability and scalability. To address this issue, we propose BIRD, a technique to detect and…

Cited by 11SourcePDFScholar
2023

Byzantine-Robust Federated Learning with Optimal Statistical Rates

AISTATS 2023poster

We propose Byzantine-robust federated learning protocols with nearly optimal statistical rates based on recent progress in high dimensional robust statistics. In contrast to prior work, our proposed protocols improve the dimension dependence and achieve a near-optimal statistical rate for strongly c…

Cited by 35SourcePDFScholar
2023

DecodingTrust: A Comprehensive Assessment of Trustworthiness in GPT Models

NeurIPS 2023oral

Generative Pre-trained Transformer (GPT) models have exhibited exciting progress in capabilities, capturing the interest of practitioners and the public alike. Yet, while the literature on the trustworthiness of GPT models remains limited, practitioners have proposed employing capable GPT models for…

2023

DensePure: Understanding Diffusion Models for Adversarial Robustness

ICLR 2023poster

Diffusion models have been recently employed to improve certified robustness through the process of denoising. However, the theoretical understanding of why diffusion models are able to improve the certified robustness is still lacking, preventing from further improvement. In this study, we close…

Cited by 43SourcePDFScholar
2023

DiffAttack: Evasion Attacks Against Diffusion-Based Adversarial Purification

NeurIPS 2023poster

Diffusion-based purification defenses leverage diffusion models to remove crafted perturbations of adversarial examples and achieve state-of-the-art robustness. Recent studies show that even advanced attacks cannot break such defenses effectively, since the purification process induces an extremely…

2022

Benchmarking Language Models for Code Syntax Understanding

EMNLP 2022finding

Pre-trained language models have demonstrated impressive performance in both natural language processing and program understanding, which represent the input as a token sequence without explicitly modeling its structure. Some prior works show that pre-trained language models can capture the syntacti…

2022

DeepStruct: Pretraining of Language Models for Structure Prediction

ACL 2022findings

We introduce a method for improving the structural understanding abilities of language models. Unlike previous approaches that finetune the models with task-specific augmentation, we pretrain language models to generate structures from the text on a collection of task-agnostic corpora. Our structure…

2022

Differentially Private Fractional Frequency Moments Estimation with Polylogarithmic Space

ICLR 2022poster

We prove that $\mathbb{F}_p$ sketch, a well-celebrated streaming algorithm for frequency moments estimation, is differentially private as is when $p\in(0, 1]$. $\mathbb{F}_p$ sketch uses only polylogarithmic space, exponentially better than existing DP baselines and only worse than the optimal non-p…

Cited by 15SourcePDFScholar
2022

Forecasting Future World Events With Neural Networks

NeurIPS 2022accept

Forecasting future world events is a challenging but valuable task. Forecasts of climate, geopolitical conflict, pandemics and economic indicators help shape policy and decision making. In these domains, the judgment of expert humans contributes to the best forecasts. Given advances in language mode…

2022

How Would The Viewer Feel? Estimating Wellbeing From Video Scenarios

NeurIPS 2022accept

In recent years, deep neural networks have demonstrated increasingly strong abilities to recognize objects and activities in videos. However, as video understanding becomes widely used in real-world applications, a key consideration is developing human-centric systems that understand not only the co…

2022

IELM: An Open Information Extraction Benchmark for Pre-Trained Language Models

EMNLP 2022main

We introduce a new open information extraction (OIE) benchmark for pre-trained language models (LM). Recent studies have demonstrated that pre-trained LMs, such as BERT and GPT, may store linguistic and relational knowledge. In particular, LMs are able to answer “fill-in-the-blank” questions when gi…

2022

Joint Language Semantic and Structure Embedding for Knowledge Graph Completion

COLING 2022main

The task of completing knowledge triplets has broad downstream applications. Both structural and semantic information plays an important role in knowledge graph completion. Unlike previous approaches that rely on either the structures or semantics of the knowledge graphs, we propose to jointly embed…

2022

PALT: Parameter-Lite Transfer of Language Models for Knowledge Graph Completion

EMNLP 2022finding

This paper presents a parameter-lite transfer learning approach of pretrained language models (LM) for knowledge graph (KG) completion. Instead of finetuning, which modifies all LM parameters, we only tune a few new parameters while keeping the original LM parameters fixed. We establish this via ref…

2022

Perturbation type categorization for multiple adversarial perturbation robustness

UAI 2022poster

Recent works in adversarial robustness have proposed defenses to improve the robustness of a single model against the union of multiple perturbation types. However, these methods still suffer significant trade-offs compared to the ones specifically trained to be robust against a single perturbation…

2022

PixMix: Dreamlike Pictures Comprehensively Improve Safety Measures

CVPR 2022poster

In real-world applications of machine learning, reliable and safe systems must consider measures of performance beyond standard test set accuracy. These other goals include out-of-distribution (OOD) robustness, prediction consistency, resilience to adversaries, calibrated uncertainty estimates, and…

Cited by 158PDFcodeScholar
2022

Scaling Out-of-Distribution Detection for Real-World Settings

ICML 2022spotlight

Detecting out-of-distribution examples is important for safety-critical machine learning applications such as detecting novel biological phenomena and self-driving cars. However, existing research mainly focuses on simple small-scale settings. To set the stage for more realistic out-of-distribution…

2021

Adversarial Examples for k-Nearest Neighbor Classifiers Based on Higher-Order Voronoi Diagrams

NeurIPS 2021poster

Adversarial examples are a widely studied phenomenon in machine learning models. While most of the attention has been focused on neural networks, other practical models also suffer from this issue. In this work, we propose an algorithm for evaluating the adversarial robustness of $k$-nearest neighbo…

2021

Aligning AI With Shared Human Values

ICLR 2021poster

We show how to assess a language model's knowledge of basic concepts of morality. We introduce the ETHICS dataset, a new benchmark that spans concepts in justice, well-being, duties, virtues, and commonsense morality. Models predict widespread moral judgments about diverse text scenarios. This requi…

2021

BACKDOORL: Backdoor Attack against Competitive Reinforcement Learning

IJCAI 2021poster

Recent research has confirmed the feasibility of backdoor attacks in deep reinforcement learning (RL) systems. However, the existing attacks require the ability to arbitrarily modify an agent's observation, constraining the application scope to simple RL systems such as Atari games. In this paper, w…

2021

Grounded Graph Decoding improves Compositional Generalization in Question Answering

EMNLP 2021finding

Question answering models struggle to generalize to novel compositions of training patterns. Current end-to-end models learn a flat input embedding which can lose input syntax context. Prior approaches improve generalization by learning permutation invariant models, but these methods do not scale to…

2021

Latent Execution for Neural Program Synthesis Beyond Domain-Specific Languages

NeurIPS 2021poster

Program synthesis from input-output (IO) examples has been a long-standing challenge. While recent works demonstrated limited success on domain-specific languages (DSL), it remains highly challenging to apply them to real-world programming languages, such as C. Due to complicated syntax and token va…

Cited by 58SourcePDFScholar
2021

Measuring Coding Challenge Competence With APPS

NeurIPS 2021poster

While programming is one of the most broadly applicable skills in modern society, it is unclear how well state-of-the-art machine learning models can write code. Despite its importance, there has been surprisingly little work on evaluating code generation, and it can be difficult to assess code gene…

Cited by 657SourcecodeScholar
2021

Measuring Massive Multitask Language Understanding

ICLR 2021poster

We propose a new test to measure a text model's multitask accuracy. The test covers 57 tasks including elementary mathematics, US history, computer science, law, and more. To attain high accuracy on this test, models must possess extensive world knowledge and problem solving ability. We find that wh…

2021

Measuring Mathematical Problem Solving With the MATH Dataset

NeurIPS 2021poster

Many intellectual endeavors require mathematical problem solving, but this skill remains beyond the capabilities of computers. To measure this ability in machine learning models, we introduce MATH, a new dataset of 12,500 challenging competition mathematics problems. Each problem in MATH has a full…

Cited by 1719SourcecodeScholar
2021

PlotCoder: Hierarchical Decoding for Synthesizing Visualization Code in Programmatic Context

ACL 2021long

Creating effective visualization is an important part of data analytics. While there are many libraries for creating visualization, writing such code remains difficult given the myriad of parameters that users need to provide. In this paper, we propose the new task of synthesizing visualization prog…

2021

Scalability vs. Utility: Do We Have To Sacrifice One for the Other in Data Importance Quantification?

CVPR 2021poster

Quantifying the importance of each training point to a learning task is a fundamental problem in machine learning and the estimated importance scores have been leveraged to guide a range of data workflows such as data summarization and domain adaption. One simple idea is to use the leave-one-out err…

Cited by 79PDFcodeScholar
2021

TeraPipe: Token-Level Pipeline Parallelism for Training Large-Scale Language Models

ICML 2021spotlight

Model parallelism has become a necessity for training modern large-scale deep language models. In this work, we identify a new and orthogonal dimension from existing model parallel approaches: it is possible to perform pipeline parallelism within a single training sequence for Transformer-based lang…

2021

The Many Faces of Robustness: A Critical Analysis of Out-of-Distribution Generalization

ICCV 2021poster

We introduce four new real-world distribution shift datasets consisting of changes in image style, image blurriness, geographic location, camera operation, and more. With our new datasets, we take stock of previously proposed methods for improving out-of-distribution robustness and put them to the t…

Cited by 2004PDFcodeScholar
2021

What Would Jiminy Cricket Do? Towards Agents That Behave Morally

NeurIPS 2021poster

When making everyday decisions, people are guided by their conscience, an internal sense of right and wrong, to behave morally. By contrast, artificial agents may behave immorally when trained on environments that ignore moral concerns, such as violent video games. With the advent of generally capab…

Cited by 69SourcecodeScholar
2021

Zero-Shot Information Extraction as a Unified Text-to-Triple Translation

EMNLP 2021main

We cast a suite of information extraction tasks into a text-to-triple translation framework. Instead of solving each task relying on task-specific datasets and models, we formalize the task as a translation between task-specific input text and output triples. By taking the task-specific input, we en…

2020

Compositional Generalization via Neural-Symbolic Stack Machines

NeurIPS 2020poster

Despite achieving tremendous success, existing deep learning models have exposed limitations in compositional generalization, the capability to learn compositional rules and apply them to unseen cases in a systematic manner. To tackle this issue, we propose the Neural-Symbolic Stack Machine (NeSS).…

Cited by 115SourcePDFScholar
2020

Neural Symbolic Reader: Scalable Integration of Distributed and Symbolic Representations for Reading Comprehension

ICLR 2020spotlight

Integrating distributed representations with symbolic operations is essential for reading comprehension requiring complex reasoning, such as counting, sorting and arithmetics, but most existing approaches are hard to scale to more domains or more complex reasoning. In this work, we propose the Neura…

Cited by 129SourceScholar
2020

Synthesize, Execute and Debug: Learning to Repair for Neural Program Synthesis

NeurIPS 2020poster

The use of deep learning techniques has achieved significant progress for program synthesis from input-output examples. However, when the program semantics become more complex, it still remains a challenge to synthesize programs that are consistent with the specification. In this work, we propose SE…

Cited by 57SourcePDFScholar
2020

The Secret Revealer: Generative Model-Inversion Attacks Against Deep Neural Networks

CVPR 2020oral

This paper studies model-inversion attacks, in which the access to a model is abused to infer information about the training data. Since its first introduction by [??], such attacks have raised serious concerns given that training data usually contain privacy sensitive information. Thus far, success…

Cited by 614PDFcodeScholar
2019

AdvIT: Adversarial Frames Identifier Based on Temporal Consistency in Videos

ICCV 2019poster

Deep neural networks (DNNs) have been widely applied in various applications, including autonomous driving and surveillance systems. However, DNNs are found to be vulnerable to adversarial examples, which are carefully crafted inputs aiming to mislead a learner to make incorrect predictions. While s…

Cited by 73PDFScholar
2019

GamePad: A Learning Environment for Theorem Proving

ICLR 2019poster

In this paper, we introduce a system called GamePad that can be used to explore the application of machine learning methods to theorem proving in the Coq proof assistant. Interactive theorem provers such as Coq enable users to construct machine-checkable proofs in a step-by-step manner. Hence, they…

2019

Synthetic Datasets for Neural Program Synthesis

ICLR 2019poster

The goal of program synthesis is to automatically generate programs in a particular language from corresponding specifications, e.g. input-output behavior. Many current approaches achieve impressive results after training on randomly generated I/O examples in limited domain-specific languages (DSLs)…

Cited by 51SourcePDFScholar
2019

Towards Efficient Data Valuation Based on the Shapley Value

AISTATS 2019poster

{\em “How much is my data worth?”} is an increasingly common question posed by organizations and individuals alike. An answer to this question could allow, for instance, fairly distributing profits among multiple data contributors and determining prospective compensation when data breaches happen. I…

Cited by 570SourcePDFScholar
2019

Using Self-Supervised Learning Can Improve Model Robustness and Uncertainty

NeurIPS 2019poster

Self-supervision provides effective representations for downstream tasks without requiring labels. However, existing approaches lag behind fully supervised training and are often not thought beneficial beyond obviating or reducing the need for annotations. We find that self-supervision can benefit r…

2018

Characterizing Adversarial Examples Based on Spatial Consistency Information for Semantic Segmentation

ECCV 2018poster

Deep Neural Networks (DNNs) have been widely applied in various recognition tasks. However, recently DNNs have been shown to be vulnerable against adversarial examples, which can mislead DNNs to make arbitrary incorrect predictions. While adversarial examples are mainly studied in classification, sp…

Cited by 121SourcePDFScholar
2018

Characterizing Adversarial Subspaces Using Local Intrinsic Dimensionality

ICLR 2018oral

Deep Neural Networks (DNNs) have recently been shown to be vulnerable against adversarial examples, which are carefully crafted instances that can mislead DNNs to make errors during prediction. To better understand such attacks, a characterization is needed of the properties of regions (the so-calle…

2018

Fooling Vision and Language Models Despite Localization and Attention Mechanism

CVPR 2018poster

Adversarial attacks are known to succeed on classifiers, but it has been an open question whether more complex vision systems are vulnerable. In this paper, we study adversarial examples for vision and language models, which incorporate natural language understanding and complex structures such as a…

2018

Improving Neural Program Synthesis with Inferred Execution Traces

NeurIPS 2018spotlight

The task of program synthesis, or automatically generating programs that are consistent with a provided specification, remains a challenging task in artificial intelligence. As in other fields of AI, deep learning-based end-to-end approaches have made great advances in program synthesis. However, mo…

Cited by 63SourcePDFScholar
2018

Parametrized Hierarchical Procedures for Neural Programming

ICLR 2018poster

Neural programs are highly accurate and structured policies that perform algorithmic tasks by controlling the behavior of a computation mechanism. Despite the potential to increase the interpretability and the compositionality of the behavior of artificial agents, it remains difficult to learn from…

Cited by 35SourcePDFScholar
2018

Practical Black-box Attacks on Deep Neural Networks using Efficient Query Mechanisms

ECCV 2018poster

Existing black-box attacks on deep neural networks (DNNs) have largely focused on transferability, where an adversarial instance generated for a locally trained model can “transfer” to attack other learning models. In this paper, we propose novel Gradient Estimation black-box attacks for adversaries…

Cited by 325SourcePDFScholar
2018

Robust Physical-World Attacks on Deep Learning Visual Classification

CVPR 2018poster

Recent studies show that the state-of-the-art deep neural networks (DNNs) are vulnerable to adversarial examples, resulting from small-magnitude perturbations added to the input. Given that that emerging physical systems are using DNNs in safety-critical situations, adversarial examples could mislea…

Cited by 2927SourcePDFScholar
2017

Delving into Transferable Adversarial Examples and Black-box Attacks

ICLR 2017poster

An intriguing property of deep neural networks is the existence of adversarial examples, which can transfer among different architectures. These transferable adversarial examples may severely hinder deep neural network-based applications. Previous works mostly study the transferability using small s…

Cited by 2183SourcecodeScholar
2016

Latent Attention For If-Then Program Synthesis

NeurIPS 2016poster

Automatic translation from natural language descriptions into programs is a long-standing challenging problem. In this work, we consider a simple yet important sub-problem: translation from textual descriptions to If-Then programs. We devise a novel neural network architecture for this task which w…

Cited by 108SourcePDFScholar