Curiosity-Diffuser: Curiosity Guide Diffusion Models for Reliability

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Liu, Zihao, Liu, Xing, Zhang, Yizhai, Liu, Zhengxiong, Huang, Panfeng
Format: Preprint
Publié: 2025
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866912282401832960
author Liu, Zihao
Liu, Xing
Zhang, Yizhai
Liu, Zhengxiong
Huang, Panfeng
author_facet Liu, Zihao
Liu, Xing
Zhang, Yizhai
Liu, Zhengxiong
Huang, Panfeng
contents One of the bottlenecks in robotic intelligence is the instability of neural network models, which, unlike control models, lack a well-defined convergence domain and stability. This leads to risks when applying intelligence in the physical world. Specifically, imitation policy based on neural network may generate hallucinations, leading to inaccurate behaviors that impact the safety of real-world applications. To address this issue, this paper proposes the Curiosity-Diffuser, aimed at guiding the conditional diffusion model to generate trajectories with lower curiosity, thereby improving the reliability of policy. The core idea is to use a Random Network Distillation (RND) curiosity module to assess whether the model's behavior aligns with the training data, and then minimize curiosity by classifier guidance diffusion to reduce overgeneralization during inference. Additionally, we propose a computationally efficient metric for evaluating the reliability of the policy, measuring the similarity between the generated behaviors and the training dataset, to facilitate research about reliability learning. Finally, simulation verify the effectiveness and applicability of the proposed method to a variety of scenarios, showing that Curiosity-Diffuser significantly improves task performance and produces behaviors that are more similar to the training data. The code for this work is available at: github.com/CarlDegio/Curiosity-Diffuser
format Preprint
id arxiv_https___arxiv_org_abs_2503_14833
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Curiosity-Diffuser: Curiosity Guide Diffusion Models for Reliability
Liu, Zihao
Liu, Xing
Zhang, Yizhai
Liu, Zhengxiong
Huang, Panfeng
Robotics
Artificial Intelligence
Machine Learning
One of the bottlenecks in robotic intelligence is the instability of neural network models, which, unlike control models, lack a well-defined convergence domain and stability. This leads to risks when applying intelligence in the physical world. Specifically, imitation policy based on neural network may generate hallucinations, leading to inaccurate behaviors that impact the safety of real-world applications. To address this issue, this paper proposes the Curiosity-Diffuser, aimed at guiding the conditional diffusion model to generate trajectories with lower curiosity, thereby improving the reliability of policy. The core idea is to use a Random Network Distillation (RND) curiosity module to assess whether the model's behavior aligns with the training data, and then minimize curiosity by classifier guidance diffusion to reduce overgeneralization during inference. Additionally, we propose a computationally efficient metric for evaluating the reliability of the policy, measuring the similarity between the generated behaviors and the training dataset, to facilitate research about reliability learning. Finally, simulation verify the effectiveness and applicability of the proposed method to a variety of scenarios, showing that Curiosity-Diffuser significantly improves task performance and produces behaviors that are more similar to the training data. The code for this work is available at: github.com/CarlDegio/Curiosity-Diffuser
title Curiosity-Diffuser: Curiosity Guide Diffusion Models for Reliability
topic Robotics
Artificial Intelligence
Machine Learning
url https://arxiv.org/abs/2503.14833