← Search

Shelby Heinecke

14 accepted papers

2026

Entropy-Based Block Pruning for Efficient Large Language Models

ICLR 2026poster

As large language models continue to scale, their growing computational and storage demands pose significant challenges for real-world deployment. In this work, we investigate redundancy within Transformer-based models and propose an entropy-based pruning strategy to enhance efficiency while maintai…

Cited by 0SourceScholar
2026

Position: Vector Prompt Interfaces Should Be Exposed to Enable Customization of Large Language Models

ICML 2026poster

As large language models (LLMs) transition from research prototypes to real-world systems, customization has emerged as a central bottleneck. While text prompts can already customize LLM behavior, we argue that text-only prompting does not constitute a suitable control interface for scalable, stable…

Cited by 0SourceScholar
2026

Test-Time Adaptation for LLM Agents via Environment Interaction

ICLR 2026poster

Large language model (LLM)-based agents struggle to generalize to novel and complex environments, such as unseen websites or new sets of functions, due to a fundamental mismatch between their pre-training and test-time conditions. This challenge stems from two distinct failure modes: a syntactic mis…

Cited by 0SourcecodeScholar
2025

APIGen-MT: Agentic Pipeline for Multi-Turn Data Generation via Simulated Agent-Human Interplay

NeurIPS 2025poster

Training effective AI agents for multi-turn interactions requires high-quality data that captures realistic human-agent dynamics, yet such data is scarce and expensive to collect manually. We introduce APIGen-MT, a two-phase framework that generates verifiable and diverse multi-turn agent data. In t…

Cited by 0SourceScholar
2025

ActionStudio: A Lightweight Framework for Data and Training of Large Action Models

EMNLP 2025

Large Action models are essential for enabling autonomous agents to perform complex tasks. However, training such models remains challenging due to the diversity of agent environments and the complexity of noisy agentic data. Existing infrastructure offers limited support for scalable, agent-specifi

2025

Diversity Empowers Intelligence: Integrating Expertise of Software Engineering Agents

ICLR 2025poster

Large language model (LLM) agents have shown great potential in solving real-world software engineering (SWE) problems. The most advanced open-source SWE agent can resolve over 27% of real GitHub issues in SWE-Bench Lite. However, these sophisticated agent frameworks exhibit varying strengths, excel…

Cited by 10SourcePDFScholar
2025

LAM SIMULATOR: Advancing Data Generation for Large Action Model Training via Online Exploration and Trajectory Feedback

ACL 2025finding

Large Action Models (LAMs) for AI Agents offer incredible potential but face challenges due to the need for high-quality training data, especially for multi-steps tasks that involve planning, executing tool calls, and responding to feedback. To address these issues, we present LAM SIMULATOR, a compr…

Cited by 0SourcePDFScholar
2025

LATTE: Learning to Think with Vision Specialists

EMNLP 2025

While open-source vision-language models perform well on simple question-answering, they still struggle with complex questions that require both perceptual and reasoning capabilities. We propose LATTE, a family of vision-language models that have LeArned to Think wiTh vision spEcialists. By offloadi

2025

PersonaBench: Evaluating AI Models on Understanding Personal Information through Accessing (Synthetic) Private User Data

ACL 2025finding

Personalization is essential for AI assistants, especially in private AI settings where models are expected to interpret users’ personal data (e.g., conversations, app usage) to understand their background, preferences, and social context. However, due to privacy concerns, existing academic research…

Cited by 23SourcePDFScholar
2025

xLAM: A Family of Large Action Models to Empower AI Agent Systems

NAACL 2025long

Autonomous agents powered by large language models (LLMs) have attracted significant research interest. However, the open-source community faces many challenges in developing specialized models for agent tasks, driven by the scarcity of high-quality agent datasets and the absence of standard protoco…

2024

APIGen: Automated PIpeline for Generating Verifiable and Diverse Function-Calling Datasets

NeurIPS 2024poster

The advancement of function-calling agent models requires diverse, reliable, and high-quality datasets. This paper presents APIGen, an automated data generation pipeline designed to synthesize high-quality datasets for function-calling applications. We leverage APIGen and collect 3,673 executable AP…

2024

Retroformer: Retrospective Large Language Agents with Policy Gradient Optimization

ICLR 2024spotlight

Recent months have seen the emergence of a powerful new trend in which large language models (LLMs) are augmented to become autonomous language agents capable of performing objective oriented multi-step tasks on their own, rather than merely responding to queries from human users. Most existing lang…

2023

Tackling Data Heterogeneity in Federated Learning with Class Prototypes

AAAI 2023technical

Data heterogeneity across clients in federated learning (FL) settings is a widely acknowledged challenge. In response, personalized federated learning (PFL) emerged as a framework to curate local models for clients' tasks. In PFL, a common strategy is to develop local and global models jointly - the…