Diffusion-Based Impedance Learning for Contact-Rich Manipulation Tasks

#### Контекст Modern методы машинного обучения достигли значительных успехов в генерации моторных действий в информационной сфере. Однако взаимодействие с физическим окружением в энергетической сфере, особенно в случае контактно-богатых задач, остается сложной проблемой. Это обусловлено тем, что традиционные подходы не ориентированы на учет физических принципов взаимодействия, а именно на них закладывается основа Impedance Control. Он требует точной настройки задачно-связанных параметров импеданса, что затрудняет применение в ситуациях, требующих высокой адаптивности. Наше исследование фокусируется на создании структуры, которая объединяет моделирование в физической сфере и методы машинного обучения для генерации точных и адаптивных импедансных контроллеров. #### Метод Мы предлагаем Diffusion-Based Impedance Learning — универсальную архитектуру, которая включает в себя следующие компоненты. Первый этап — использование Transformer-based Diffusion Model, который обучается по данным симуляционных нулевых силовых траекторий (sZFT) с применением cross-attention к внешним векторам сил. Это позволяет модели генерировать в точности движение в трансляционном и вращательном пространствах. Для решения проблемы геометрической консистентности во вращательном пространстве мы вводим SLERP-based quaternion noise scheduler. Второй этап — обновление параметров импеданса (стиффнесс и демпфинг) с помощью energy-based estimator. Этот этап включает в себя дирекционный подход: контроллер уменьшает импеданс вдоль незадачных осей, оставляя высокую резистивность вдоль задачных направлений. Мы собирали данные в сценарии паркура и робототехнической терапии с помощью Apple Vision Pro. #### Результаты Проведенные эксперименты показали высокую эффективность метода. На данных, собранных в сценарии паркура, модель достигла суб-миллиметровой точности позиции и суб-динамической точности подвижных роботов. Модель показала успех в решении задачи подгонки в контактных контактовых задачах, в том числе поворотов и вставки эллиптических и цилиндрических точек, без предварительных демонстраций в обучающих данных. Это достигнуто благодаря возможности автоматической адаптации импеданса во время выполнения задачи. Уменьшенный размер модели позволил применять этот подход в реальном времени, в том числе с автоматическим корректированием динамики на KUKA LBR iiwa. #### Значимость Предложенный подход может быть применен в различных контактно-богатых задачах, в том числе в робототехнических терапиях, производственных процессах и распределенных системах с удаленным управлением. Он обеспечивает высокую точность и реактивность, объединяя моде