← Search

Zichong Li

11 accepted papers

2026

COSMOS: A Hybrid Adaptive Optimizer for Efficient Training of Large Language Models

ICLR 2026poster

Large Language Models (LLMs) have demonstrated remarkable success across various domains, yet their optimization remains a significant challenge due to the complex and high-dimensional loss landscapes they inhabit. While adaptive optimizers such as AdamW are widely used, they suffer from critical li…

Cited by 0SourcecodeScholar
2026

Shuffle the Context: RoPE-Perturbed Self-Distillation for Long-Context Adaptation

ICML 2026poster

Large language models (LLMs) increasingly operate in settings that require reliable long-context understanding, such as retrieval-augmented generation and multi-document reasoning. A common strategy is to fine-tune pretrained short-context models at the target sequence length. However, we find that …

Cited by 0SourceScholar
2025

MARLadona - Towards Cooperative Team Play Using Multi-Agent Reinforcement Learning

ICRA 2025

Robot soccer, in its full complexity, poses an unsolved research challenge. Current solutions heavily rely on engineered heuristic strategies, which lack robustness and adaptability. Deep reinforcement learning has gained significant traction in various complex robotics tasks such as locomotion, man

Cited by 11SourceScholar
2024

Adaptive Preference Scaling for Reinforcement Learning with Human Feedback

NeurIPS 2024poster

Reinforcement learning from human feedback (RLHF) is a prevalent approach to align AI systems with human values by learning rewards from human preference data. Due to various reasons, however, such data typically takes the form of rankings over pairs of trajectory segments, which fails to capture th…

Cited by 5SourcePDFScholar
2024

Beyond Point Prediction: Score Matching-based Pseudolikelihood Estimation of Neural Marked Spatio-Temporal Point Process

ICML 2024poster

Spatio-temporal point processes (STPPs) are potent mathematical tools for modeling and predicting events with both temporal and spatial features. Despite their versatility, most existing methods for learning STPPs either assume a restricted form of the spatio-temporal distribution, or suffer from in…

Cited by 3SourcePDFScholar
2024

Robust Reinforcement Learning from Corrupted Human Feedback

NeurIPS 2024poster

Reinforcement learning from human feedback (RLHF) provides a principled framework for aligning AI systems with human preference data. For various reasons, e.g., personal bias, context ambiguity, lack of training, etc, human annotators may give incorrect or inconsistent preference labels. To tackle…

Cited by 4SourcePDFScholar
2023

SMURF-THP: Score Matching-based UnceRtainty quantiFication for Transformer Hawkes Process

ICML 2023poster

Transformer Hawkes process models have shown to be successful in modeling event sequence data. However, most of the existing training methods rely on maximizing the likelihood of event sequences, which involves calculating some intractable integral. Moreover, the existing methods fail to provide unc…

2022

Zeroth-Order Optimization for Composite Problems with Functional Constraints

AAAI 2022technical

In many real-world problems, first-order (FO) derivative evaluations are too expensive or even inaccessible. For solving these problems, zeroth-order (ZO) methods that only need function evaluations are often more efficient than FO methods or sometimes the only options. In this paper, we propose a n…

Cited by 7SourcePDFScholar
2021

Rate-improved inexact augmented Lagrangian method for constrained nonconvex optimization

AISTATS 2021poster

First-order methods have been studied for nonlinear constrained optimization within the framework of the augmented Lagrangian method (ALM) or penalty method. We propose an improved inexact ALM (iALM) and conduct a unified analysis for nonconvex problems with either convex or nonconvex constraints. U…

Cited by 62SourcePDFScholar