← Search

Jiaji Zhang

7 accepted papers

2026

SegQuant: A Semantics-Aware and Generalizable Quantization Framework for Diffusion Models

CVPR 2026

Diffusion models have demonstrated exceptional generative capabilities but are computationally intensive, posing significant challenges for deployment in resource-constrained or latency-sensitive environments.Quantization offers an effective means to reduce model size and computational cost, with po

Cited by 0SourcecodeScholar
2025

Any-step Dynamics Model Improves Future Predictions for Online and Offline Reinforcement Learning

ICLR 2025poster

Model-based methods in reinforcement learning offer a promising approach to enhance data efficiency by facilitating policy exploration within a dynamics model. However, accurately predicting sequential steps in the dynamics model remains a challenge due to the bootstrapping prediction, which attribu…

2025

Robustifying Learning-Augmented Caching Efficiently without Compromising 1-Consistency

NeurIPS 2025poster

The online caching problem aims to minimize cache misses when serving a sequence of requests under a limited cache size. While naive learning-augmented caching algorithms achieve ideal $1$-consistency, they lack robustness guarantees. Existing robustification methods either sacrifice $1$-consistency…

Cited by 0SourceScholar
2024

Episodic Return Decomposition by Difference of Implicitly Assigned Sub-trajectory Reward

AAAI 2024technical

Real-world decision-making problems are usually accompanied by delayed rewards, which affects the sample efficiency of Reinforcement Learning, especially in the extremely delayed case where the only feedback is the episodic reward obtained at the end of an episode. Episodic return decomposition is a…

2024

Flow to Better: Offline Preference-based Reinforcement Learning via Preferred Trajectory Generation

ICLR 2024poster

Offline preference-based reinforcement learning (PbRL) offers an effective solution to overcome the challenges associated with designing rewards and the high costs of online interactions. In offline PbRL, agents are provided with a fixed dataset containing human preferences between pairs of trajecto…

Cited by 20SourcePDFScholar
2024

KALM: Knowledgeable Agents by Offline Reinforcement Learning from Large Language Model Rollouts

NeurIPS 2024poster

Reinforcement learning (RL) traditionally trains agents using interaction data, which limits their capabilities to the scope of the training data. To create more knowledgeable agents, leveraging knowledge from large language models (LLMs) has shown a promising way. Despite various attempts to combin…

2023

Model-Bellman Inconsistency for Model-based Offline Reinforcement Learning

ICML 2023poster

For offline reinforcement learning (RL), model-based methods are expected to be data-efficient as they incorporate dynamics models to generate more data. However, due to inevitable model errors, straightforwardly learning a policy in the model typically fails in the offline setting. Previous studies…