← Search

Jiaxi Yang

23 accepted papers

2026

Benchmarking the Limits of In-Context Reinforcement Learning for Ad-Hoc Teamwork

ICML 2026poster

In-Context Reinforcement Learning (ICRL) has enabled foundation agents to adapt instantaneously to novel tasks, yet its efficacy in Ad-Hoc Teamwork (AHT)—where coordination with unknown partners is required—remains unexplored. To rigorously evaluate this, we introduce a large-scale benchmark **ICRL4…

Cited by 0SourceScholar
2026

Bringing Code ALIVE: Optimizing Interactive Frontend Mini-Games via Automated Play and Reinforcement Learning at Scale

ICML 2026poster

The rapid evolution of Large Language Models (LLMs) has empowered even non-programmers to create visually appealing frontend mini-games with a single instruction. However, open-source models significantly lag behind proprietary counterparts in this domain. The core bottleneck is the lack of an evalu…

Cited by 0SourceScholar
2026

PlotCraft: Pushing the Limits of LLMs for Complex and Interactive Data Visualization

ICML 2026poster

Recent Large Language Models (LLMs) have demonstrated remarkable proficiency in code generation. However, their ability to create complex visualizations for scaled and structured data remains largely unevaluated and underdeveloped. To address this gap, we introduce **PlotCraft**, a new benchmark fea…

Cited by 0SourceScholar
2026

SWE-RM: Execution-free Feedback for Software Engineering Agents

ICLR 2026poster

Execution-based feedback like unit testing is widely used in the development of coding agents through test-time scaling (TTS) and reinforcement learning (RL). This paradigm requires scalable and reliable collection of unit test cases to provide accurate feedback, and the resulting feedback is often…

Cited by 0SourceScholar
2026

Scaling Agentic Verifier for Competitive Coding

ICML 2026poster

Large language models (LLMs) have demonstrated strong coding capabilities but still struggle to solve competitive programming problems correctly in a single attempt. Execution-based re-ranking offers a promising test-time scaling strategy, yet existing methods are constrained by either difficult tes…

Cited by 0SourceScholar
2025

CodeArena: Evaluating and Aligning CodeLLMs on Human Preference

EMNLP 2025

We present CodeArena to emulate the complexity/diversity of real-world coding tasks, spanning 40 categories and 44 PLs. A 20B diverse synthetic instruction corpus is created by scaling instructions to help Qwen2.5-SynCoder achieve SOTA performance. Abstract: Code large language models (codeLLMs) hav

Cited by 0SourcePDFScholar
2025

Do Vision-Language Models Have Internal World Models? Towards an Atomic Evaluation

ACL 2025finding

Internal world models (WMs) enable agents to understand the world’s state and predict transitions, serving as the basis for advanced deliberative reasoning.Recent large Vision-Language Models (VLMs), such as GPT-4o and Gemini, exhibit potential as general-purpose WMs. While the latest studies have e…

Cited by 0SourcePDFScholar
2025

Fine-Tuning Language Models with Collaborative and Semantic Experts

AAAI 2025technical

Recent advancements in large language models (LLMs) have broadened their application scope but revealed challenges in balancing capabilities across general knowledge, coding, and mathematics. To address this, we introduce a Collaborative and Semantic Experts (CoE) approach for supervised fine-tuning…

Cited by 0SourcePDFScholar
2025

GMValuator: Similarity-based Data Valuation for Generative Models

ICLR 2025poster

Data valuation plays a crucial role in machine learning. Existing data valuation methods, mainly focused on discriminative models, overlook generative models that have gained attention recently. In generative models, data valuation measures the impact of training data on generated datasets. Very few…

2025

Hierarchical Context Pruning: Optimizing Real-World Code Completion with Repository-Level Pretrained Code LLMs

AAAI 2025technical

Some of the latest released Code Large Language Models (Code LLMs) have been trained on repository-level code data, enabling them to perceive repository structures and utilize cross-file code information. This capability allows us to directly concatenate the content of repository code files in promp…

2025

Parallel Scaling Law for Language Models

NeurIPS 2025poster

It is commonly believed that scaling language models should commit a significant space or time cost, by increasing the parameters (parameter scaling) or output tokens (inference-time scaling). We introduce another and more inference-efficient scaling paradigm: increasing the model's parallel computa…

Cited by 0SourcecodeScholar
2025

START: Self-taught Reasoner with Tools

EMNLP 2025

Large Reasoning Models (LRMs) have demonstrated remarkable capabilities in complex reasoning through long chain-of-thought, yet they struggle with precise computations and algorithmic operations. Integrating computational tools with LRMs remains challenging, particularly in activating and enhancing

2025

Separating the Wheat from the Chaff: Spatio-Temporal Transformer with View-interweaved Attention for Photon-Efficient Depth Sensing

AAAI 2025technical

Time-resolved imaging is an emerging sensing modality that has been shown to enable advanced applications, including remote sensing, fluorescence lifetime imaging, and even non-line-of-sight sensing. Single-photon avalanche diodes (SPADs) outperform relevant time-resolved imaging technologies thanks…

Cited by 0SourcePDFScholar
2025

Synthesizing Software Engineering Data in a Test-Driven Manner

ICML 2025poster

We introduce **SWE-Flow**, a novel data synthesis framework grounded in Test-Driven Development (TDD). Unlike existing software engineering data that rely on human-submitted issues, **SWE-Flow** automatically infers incremental development steps directly from unit tests, which inherently encapsulate…

2025

Turning the Tide: Repository-based Code Reflection

EMNLP 2025

Code large language models (LLMs) enhance programming by understanding and generating code across languages, offering intelligent feedback, bug detection, and code updates through reflection, improving development efficiency and accessibility. While benchmarks (e.g. HumanEval/LiveCodeBench) evaluate

2024

Iterative Forward Tuning Boosts In-Context Learning in Language Models

ACL 2024long

Despite the advancements in in-context learning (ICL) for large language models (LLMs), current research centers on specific prompt engineering, such as demonstration selection, with the expectation that a single iteration of demonstrations processing can generalize effectively to a given test sampl…

2024

Marathon: A Race Through the Realm of Long Context with Large Language Models

ACL 2024long

With the advancement of large language models (LLMs) and the expansion of their context windows, existing long-context benchmarks fall short in effectively evaluating the models’ comprehension and reasoning abilities in extended texts. Moreover, conventional benchmarks relying on F1 metrics often in…

2024

One-Shot Learning as Instruction Data Prospector for Large Language Models

ACL 2024long

Contemporary practices in instruction tuning often hinge on enlarging data scaling without a clear strategy for ensuring data quality, inadvertently introducing noise that may compromise model performance. To address this challenge, we introduce Nuggets, a novel and efficient methodology that levera…

2024

Synthesizing Text-to-SQL Data from Weak and Strong LLMs

ACL 2024long

The capability gap between open-source and closed-source large language models (LLMs) remains a challenge in text-to-SQL tasks. In this paper, we introduce a synthetic data approach that combines data produced by larger, more powerful models (strong models) with error information data generated by s…

2023

Can LLM Already Serve as A Database Interface? A BIg Bench for Large-Scale Database Grounded Text-to-SQLs

NeurIPS 2023spotlight

Text-to-SQL parsing, which aims at converting natural language instructions into executable SQLs, has gained increasing attention in recent years. In particular, GPT-4 and Claude-2 have shown impressive results in this task. However, most of the prevalent benchmarks, i.e., Spider, and WikiSQL, focu…

2022

Glass Segmentation Using Intensity and Spectral Polarization Cues

CVPR 2022poster

Transparent and semi-transparent materials pose significant challenges for existing scene understanding and segmentation algorithms due to their lack of RGB texture which impedes the extraction of meaningful features. In this work, we exploit that the light-matter interactions on glass materials pro…

Cited by 93PDFScholar
2021

A User-Adaptive Layer Selection Framework for Very Deep Sequential Recommender Models

AAAI 2021technical

Sequential recommender systems (SRS) have become a research hotspot in recent studies. Because of the requirement in capturing user's dynamic interests, sequential neural network based recommender models often need to be stacked with more hidden layers (e.g., up to 100 layers) compared with standard…

Cited by 12SourcePDFScholar
2021

Differentially Private Correlation Alignment for Domain Adaptation

IJCAI 2021poster

Domain adaptation solves a learning problem in a target domain by utilizing the training data in a different but related source domain. As a simple and efficient method for domain adaptation, correlation alignment transforms the distribution of the source domain by utilizing the covariance matrix of…

Cited by 7SourcePDFScholar