Limits of imitation learning for high-precision insertion

BSc thesis · ACT and Diffusion Policy on 30k simulated demos (UR5e, Isaac Lab)

BSc thesis, ETF Robotics Laboratory, University of Belgrade (2026). Advisor: Prof. Kosta Jovanović. Code · Policies on Hugging Face · Eval results

How far can imitation learning go on a task that needs millimetre precision? I studied this on fiber-optic (SFP) connector insertion with a UR5e arm in NVIDIA Isaac Lab.

  • Built the task environment (AIC-Port-Insertion-v0) and a scripted oracle that generated 30,290 successful expert demonstrations.
  • Trained ACT and Diffusion Policy (LeRobot) with an auxiliary phase-prediction head, at 1k, 10k and 30k demonstrations.
  • Evaluated closed-loop over 150 episodes per policy, with automatic failure classification.

Findings. The main obstacle is lateral error, the sideways misalignment with the port. More data helps a lot: Diffusion Policy’s median lateral error dropped 3.5× (41.2 → 11.8 mm), but that is still far from the millimetre precision the insertion requires.

A successful Diffusion Policy episode.