Hybrid Diffusion Policies with Projective Geometric Algebra for Efficient Robot Manipulation Learning

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Sun, Xiatao, Wang, Yuxuan, Yang, Shuo, Chen, Yinxing, Rakita, Daniel
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866917319905640448
author Sun, Xiatao
Wang, Yuxuan
Yang, Shuo
Chen, Yinxing
Rakita, Daniel
author_facet Sun, Xiatao
Wang, Yuxuan
Yang, Shuo
Chen, Yinxing
Rakita, Daniel
contents Diffusion policies are a powerful paradigm for robot learning, but their training is often inefficient. A key reason is that networks must relearn fundamental spatial concepts, such as translations and rotations, from scratch for every new task. To alleviate this redundancy, we propose embedding geometric inductive biases directly into the network architecture using Projective Geometric Algebra (PGA). PGA provides a unified algebraic framework for representing geometric primitives and transformations, allowing neural networks to reason about spatial structure more effectively. In this paper, we introduce hPGA-DP, a novel hybrid diffusion policy that capitalizes on these benefits. Our architecture leverages the Projective Geometric Algebra Transformer (P-GATr) as a state encoder and action decoder, while employing established U-Net or Transformer-based modules for the core denoising process. Through extensive experiments and ablation studies in both simulated and real-world environments, we demonstrate that hPGA-DP significantly improves task performance and training efficiency. Notably, our hybrid approach achieves substantially faster convergence compared to both standard diffusion policies and architectures that rely solely on P-GATr. The project website is available at: https://apollo-lab-yale.github.io/26-ICRA-hPGA-website/.
format Preprint
id arxiv_https___arxiv_org_abs_2507_05695
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Hybrid Diffusion Policies with Projective Geometric Algebra for Efficient Robot Manipulation Learning
Sun, Xiatao
Wang, Yuxuan
Yang, Shuo
Chen, Yinxing
Rakita, Daniel
Robotics
Diffusion policies are a powerful paradigm for robot learning, but their training is often inefficient. A key reason is that networks must relearn fundamental spatial concepts, such as translations and rotations, from scratch for every new task. To alleviate this redundancy, we propose embedding geometric inductive biases directly into the network architecture using Projective Geometric Algebra (PGA). PGA provides a unified algebraic framework for representing geometric primitives and transformations, allowing neural networks to reason about spatial structure more effectively. In this paper, we introduce hPGA-DP, a novel hybrid diffusion policy that capitalizes on these benefits. Our architecture leverages the Projective Geometric Algebra Transformer (P-GATr) as a state encoder and action decoder, while employing established U-Net or Transformer-based modules for the core denoising process. Through extensive experiments and ablation studies in both simulated and real-world environments, we demonstrate that hPGA-DP significantly improves task performance and training efficiency. Notably, our hybrid approach achieves substantially faster convergence compared to both standard diffusion policies and architectures that rely solely on P-GATr. The project website is available at: https://apollo-lab-yale.github.io/26-ICRA-hPGA-website/.
title Hybrid Diffusion Policies with Projective Geometric Algebra for Efficient Robot Manipulation Learning
topic Robotics
url https://arxiv.org/abs/2507.05695