PAC-ACT: Post-training Actor-Critic for Action Chunking Transformers

· Source: Artificial Intelligence · Field: Technology & Digital — Robotics & Autonomous Systems, Artificial Intelligence & Machine Learning · Depth: Expert, quick

Summary

PAC-ACT is a novel reinforcement-learning post-training framework designed for pretrained Action Chunking Transformer (ACT) policies, addressing challenges in precision industrial contact manipulation. While vision-language-action models offer broad generalization, they often incur high inference latency and GPU-memory costs. Existing vision-action chunking policies, typically trained via behavior cloning, struggle with distribution shift in contact-rich tasks. PAC-ACT tackles this by reformulating policy optimization at the chunk level, constructing an ACT-transferred actor-critic architecture, and integrating a hybrid behavior-prior constraint to maintain the pretrained action distribution during online fine-tuning. Experimental results on industrial precision-contact benchmarks demonstrate that PAC-ACT enhances task success, contact stability, and force safety, all while preserving low latency and GPU-memory usage. Notably, on the Contour task, it significantly reduces peak contact force and decreases the proportion of force readings above 60 N by 46 times. Sparse-reward ablations confirm the behavior-prior constraint's effectiveness in exploration under randomized initial poses.

Key takeaway

For Robotics Engineers developing policies for precision industrial contact manipulation, PAC-ACT provides a robust post-training solution. If your current behavior cloning-based Action Chunking Transformer policies exhibit distribution shift or instability in contact-rich tasks, you should consider this reinforcement learning framework. It can significantly improve task success, contact stability, and force safety, reducing peak contact forces by 46 times in some cases, while maintaining low latency and GPU-memory usage for real-time industrial control.

Key insights

PAC-ACT improves robot policy reliability in contact-rich tasks by post-training Action Chunking Transformers with RL and a behavior-prior constraint.

Principles

Method

PAC-ACT reformulates policy optimization at the chunk level, builds an ACT-transferred actor-critic, and uses a hybrid behavior-prior constraint for online fine-tuning.

In practice

Topics

Best for: Research Scientist, Robotics Engineer, AI Scientist

Related on AIssential

Open in AIssential →

Editorial summary, takeaway, and curation by AIssential. Original article published by Artificial Intelligence.