CelloAI: Leveraging Large Language Models for HPC Software Development in High Energy Physics

## Контекст High Energy Physics (HEP) находится на переломном этапе развития, возникающем в связи с ростом требований к обработке и хранению данных, полученных в результате высокочастотных экспериментов. Этот бурный рост данных приводит к необходимости внедрения высокопроизводительных вычислений (HPC) в существующую инфраструктуру, в которой возникают новые сложности, такие как необходимость портирования устаревшего программного обеспечения на новые архитектуры и недостаточность документации, связанной с характером высокотехнологичных научных приложений. Эти проблемы ограничивают возможности эффективного использования HPC-технологий в HEP. Для решения этих задач необходим развитый инструментарий, который может упростить процессы документирования и генерации кода, а также помочь в понимании и модификации сложных научных приложений. ## Метод CelloAI — это локально развернутый кодинговый помощник, основанный на Large Language Models (LLMs), который использует технологию Retrieval-Augmented Generation (RAG) для поддержки документации и генерации кода в области теоретической физики. Архитектура CelloAI включает в себя несколько специализированных компонентов для различных задач. Для документации кода он предлагает: (a) генерацию комментариев в стиле Doxygen для функций и классов, (b) генерацию сводки по файлам, и (c) взаимодействие с интерактивным чатботом для запросов по пониманию кода. Для генерации кода CelloAI использует синтаксически ориентированные алгоритмы, которые сохраняют синтаксические границы во время задания входных данных, что улучшает точность восстановления в больших кодовых базах. Он также использует связанные с графами вызовов информацию для сохранения связности во время изменений. Локальное развертывание CelloAI обеспечивает безопасность данных, экономичность и доступ к большим контекстным окружениям без зависимостей от внешних систем. ## Результаты Чтобы оценить эффективность CelloAI, мы провели эксперименты с реальными приложениями HEP, в том числе с приложениями ATLAS, CMS и DUNE. Мы сравнили различные модели вставки для кода в целях выбора наиболее эффективной. Эксперименты показали, что восстановление кода с помощью CelloAI значительно улучшает понимание и поддержку кода. Также были протестированы возможности генерации кода, в том числе предложений для оптимизации производительности и рефакторинга. Результаты показали, что CelloAI обеспечивает надежную поддержку разработки кода, обеспечивая при этом полную прозрачность и безопасность, необходимые для научных приложений. ## Значимость CelloAI может быть применен в различных областях, где требуется поддержка генерации и документации кода в сложных науч