Impact-driven Context Filtering For Cross-file Code Completion

#### Контекст Инструменты автоматической подсказки кода, такие как cross-file code completion, становятся важной частью процесса разработки программного обеспечения. Эти инструменты используются для ускорения разработки, уменьшения числа ошибок и улучшения производительности разработчиков. Однако, существуют проблемы, связанные с точностью и эффективностью подсказок. Одна из них — неточность в выборе и использовании контекста из других файлов. Это приводит к нежелательным рекомендациям и ухудшению качества выдачи. Необходимо развитие методов, которые позволят более точно определить и использовать полезные части контекста. #### Метод Мы предлагаем новую методологию, основанную на ретриев-аугментированной генерации (RAG), которая объединяет в себе контекст из вложенных файлов и ранее написанного кода. Мы используем возможности глубокого обучения для построения модели, которая может оценивать вклад каждого отдельного фрагмента кода в общую точность рекомендации. Для этого мы разработали метрику, основанную на вероятностях, для оценки того, насколько каждый фрагмент кода полезен для рекомендации. Мы также создали новый датасет, где каждый фрагмент помечен как полезный (positive), нейтральный (neutral) или вредный (negative). #### Результаты Мы провели эксперименты на двух открытых бенчмарках: RepoEval и CrossCodeLongEval. Модель CODEFILTER, которая основывается на нашем подходе, показала существенное улучшение точности рекомендаций в сравнении с моделями, не использующими фильтрацию контекста. Мы также отметили, что модель CODEFILTER уменьшает длину входного последовательности (prompt), что улучшает эффективность вычислений. Мы также отметили, что наш подход обладает значительной генерализуемостью и работает хорошо со многими моделями различных архитектур. #### Значимость Наш подход может быть применен в различных сферах, где требуется точное и эффективное подсказывание кода. Например, в системах автоматической поддержки разработчиков, системах анализа кода и системах помощи в написании кода. Он позволяет улучшить точность и эффективность кода, а также обеспечивает более четкую атрибуцию каждого фрагмента кода, что может помочь в исследовании и улучшении качества моделей. #### Выводы Мы установили, что наш подход приводит к значительным улучшениям в точности рекомендации и эффективности вычислений в cross-file code completion. Будущие исследования будут сфокусированы на улучшении принципов фильтрации контекста, на расширении области применения модели и на дополнительной оптимизации производительности. Мы планируем также расширить наш датасет, чтобы включить более разнообразные задачи и задачу с многомодельным подходом.