Разработчик Шив Шанмугам (Shiv Shanmugam) представил Taiga‑S1 — компактную модель с 1,2 млн параметров, которая может создавать 3D‑детали во FreeCAD. Нейросеть принимает решения примерно за 1 мс. При этом ей не нужны LLM, скриншоты интерфейса и компьютерное зрение.
Taiga‑S1 задумана как быстрый слой для более крупного AI‑агента. Планировщик определяет, какую делать надо построить, и передаёт структурированный список операций. После этого Taiga‑S1 выбирает, какие команды FreeCAD нужно для этого использовать: создать эскиз, выдавить область, сделать отверстие, снять фаску и так далее. На каждом шаге модель получает текущее состояние объекта и выбирает следующую допустимую команду. Числовые параметры при этом она не генерирует. Они приходят из исходного задания.
Модель обучили с нуля на 25 тыс. синтетических сессий моделирования, включающих около 590 тыс. решений. После базового обучения автор модели провёл ещё два этапа дообучения с помощью итеративного сбора и добавления примеров ошибок модели: Taiga‑S1 самостоятельно управляла FreeCAD, а программный учитель исправлял её ошибки. Архитектура модели построена на трёхслойном трансформере.
В тестах Taiga‑S1 выполнила 100% заданий с 11 элементами. Для каждого потребовалось сделать 55 действий, хотя во время обучения нейросеть видела последовательности максимум с пятью элементами. С заданиями из 17 элементов модель справилась в 95% случаев.
Код Taiga‑S1 опубликован на GitHub под лицензией MIT, а веса доступны на Hugging Face.
Источник: habr.com