Bi-MoDe: Bilateral Control-based Imitation Learning via Modifier-Conditioned Decoding for Modulation of Execution Speed and Contact Intensity
Organizations: Graduate School of Information Science and Technology, The University of Osaka · D3 Center, The University of Osaka · Graduate School of Maritime Sciences, Kobe University
Abstract
Bilateral control-based imitation learning captures both position and force information, making it well suited to contact-rich manipulation. However, existing approaches provide limited means for an operator to specify how a learned task should be executed at inference time, such as slowly or quickly, gently or firmly. We propose Bi-MoDe, a modifier-conditioned decoding framework that injects a constrained latent into every layer of the Transformer action decoder via adaLN-Zero, allowing behavioral directives to directly influence action-chunk generation. We evaluate the method on a real-world whiteboard wiping task with combinations of temporal and physical modifiers. Bi-MoDe improves physical directive following over the action-chunking baseline while maintaining comparable temporal control. An ablation further shows that decoder conditioning and latent-space composition interact, and that their combination is important for accurate physical directive following. Additional material is available at the https://mertcookimg.github.io/bi-mode/
Figures & tables
| Parameter | Value |
|---|---|
| Encoder layers | 4 |
| Decoder layers | 4 |
| Hidden dimension | 512 |
| Feedforward dimension | 2048 |
| Attention heads | 8 |
| Latent dimension ( / ) | 2 / 1 |
| Configuration | Design | TSR | Physical | Temporal | |||
|---|---|---|---|---|---|---|---|
| Decoder cond. | MDE | slope | MDE | slope | |||
| Baseline (ACT-based) [ 10 ] | – | ✓ | 45/45 | 0.123 | 0.896 | 0.161 | 1.104 |
| No | – | – | 45/45 | 0.300 | 0.754 | 0.093 | 1.069 |
| Decoder conditioned | ✓ | ✓ | 45/45 | 0.139 | 0.854 | 0.039 | 0.959 |
| Bi-MoDe (Ours) | ✓ | – | 45/45 | 0.036 | 0.968 | 0.116 | 1.096 |