From Intent to Execution: Multimodal Chain-of-Thought Reinforcement Learning for Precise CAD Code Generation

#### Контекст Машинно-управляемый дизайн (CAD) широко используется в инженерном проектировании и производстве, где требуется точность и компетентность. Однако существующие CAD-системы часто требуют значительного времени и усилий для моделирования, так как они зависят от экспертного знания и ручных операций. Искусственные нейросетевые модели, особенно те, которые используют технологии типа LLMs, стали помогать автоматизировать процессы, но в случае CAD-моделирования существуют ряд проблем: требования к логическому рассуждению, корректности синтаксиса и точности значений. Большинство существующих подходов столкнулись с трудностями в создании решений, которые бы учитывали все эти аспекты одновременно. Наша цель — разработать метод, который бы решал эти проблемы и позволял эффективно генерировать код CAD на основе естественного языка. #### Метод Мы предлагаем **CAD-RL**, фреймворк, основанный на Chain-of-Thought (CoT) и работающий на основе усовершенствованного пост-тренировочного развития с помощью эвристических подходов. Наша методика включает несколько ключевых элементов: 1. **Multimodal Chain-of-Thought (CoT):** мы используем CoT для обеспечения логического рассуждения в ходе моделирования. 2. **Goal-driven reinforcement learning (RL) post-training:** наша подходящая тренировка позволяет менять поведение модели в соответствии с целями решения задач. 3. **Заданные награды:** включаются три основных вида наград для тренировочного процесса: награда за выполнение, награда за точность геометрии и награда за внешнюю оценку. 4. **Оптимизационные стратегии:** включаются Trust Region Stretch для лучшего исследования, Precision Token Loss для улучшения точности параметров размеров, и Overlong Filtering для борьбы с шумом в наблюдениях. 5. **Dataset ExeCAD:** мы выпустили новую базу данных с реальными примерами CAD, включающую 16,540 задач, с решениями в виде естественного языка, запросов на моделирование и CADQuery-скриптами. #### Результаты Мы проверили CAD-RL на ExeCAD, сравнив его с различными нынешними подходами. Наши результаты показали, что CAD-RL добивается значительного улучшения качества моделей, в частности, превышает другие методы в логической точности, точности геометрии и эффективности выполнения. Мы также провели эксперименты, которые подтвердили, что наша технология выдает лучший результат в плане качества интерпретации и выполнения кода. #### Значимость Наш подход может применяться в различных областях, включая автоматизацию процессов CAD, создание современных продуктов, улучшение технологий инженерии. CAD-RL дает более точные и надежные решения, что может ускорить рабочий процесс и уменьшить человеческую о