← Search

Weichen Li

7 accepted papers

2026

Landmark-Guided Policy Optimization for Multi-Objective Language Model Selection

ICML 2026poster

Selecting a pretrained large language model (LLM) to fine-tune for a task-specific dataset can be time-consuming and costly. With several candidate models available to choose from, varying in size, architecture, and pretraining data, finding the best model for a specific task often involves extensiv…

Cited by 0SourceScholar
2026

SemRep: Code Transformation with Semantics-Preserving Representations

ICML 2026poster

Code transformation is a foundational capability in the software development process, where its effectiveness relies on constructing a high-quality code representation to characterize the input code semantics and guide the transformation. Existing approaches treat code transformation as an end-to-en…

Cited by 0SourceScholar
2026

TORA: Train Once, Realign Anytime for Offline Multi-Objective Reinforcement Learning

AAAI 2026technical

Intelligent agents in real-world applications must adapt their behavior to changing contexts and user preferences. For example, planning a road trip requires considering both travel time and cost. Multi-objective reinforcement learning (MORL) provides a principled approach to navigate such trade-of

Cited by 0SourcePDFScholar
2025

EditLord: Learning Code Transformation Rules for Code Editing

ICML 2025poster

Code editing is a foundational task in software development, where its effectiveness depends on whether it introduces desired code property changes without changing the original code's intended functionality. Existing approaches often formulate code editing as an implicit end-to-end task, omitting…

Cited by 0SourcePDFScholar
2024

Beyond Entities: A Large-Scale Multi-Modal Knowledge Graph with Triplet Fact Grounding

AAAI 2024technical

Much effort has been devoted to building multi-modal knowledge graphs by visualizing entities on images, but ignoring the multi-modal information of the relation between entities. Hence, in this paper, we aim to construct a new large-scale multi-modal knowledge graph with triplet facts grounded on i…

2024

Ethics in Action: Training Reinforcement Learning Agents for Moral Decision-making In Text-based Adventure Games

AISTATS 2024poster

Reinforcement Learning (RL) has demonstrated its potential in solving goal-oriented sequential tasks. However, with the increasing capabilities of RL agents, ensuring morally responsible agent behavior is becoming a pressing concern. Previous approaches have included moral considerations by statical…

2024

Exploiting Code Symmetries for Learning Program Semantics

ICML 2024spotlight

This paper tackles the challenge of teaching code semantics to Large Language Models (LLMs) for program analysis by incorporating code symmetries into the model architecture. We introduce a group-theoretic framework that defines code symmetries as semantics-preserving transformations, where forming…

Cited by 7SourcePDFScholar