zkLoRA: Fine-Tuning Large Language Models with Verifiable Security via Zero-Knowledge Proofs

## Контекст Fine-tuning больших языковых моделей (LLMs) является ключевым инструментом для их адаптации к конкретным задачам, однако этот процесс требует огромных вычислительных ресурсов и порождает затруднения в удостоверении правильности результатов, особенно в небезопасных средах. Дополнительно, вопросы конфиденциальности моделей и данных, особенно в условиях доверия к стороне, остаются неразрешимыми. Известные методы, такие как Low-Rank Adaptation (LoRA), уменьшают вычислительные затраты, но не обеспечивают гарантии безопасности и проверяемости. Наша исследовательская цель состоит в разработке безопасной системы для проверки и защиты LoRA-based fine-tuning, учитывая требования к конфиденциальности и проверяемости в архитектурах Transformer-based. ## Метод вводится zkLoRA — первый фреймворк, комбинирующий LoRA-файнтюнинг с технологией zero-knowledge proofs (ZKP). Мы используем высокотехнологичные криптографические методы, такие как lookup-аргументы, sumcheck-протоколы и polynomial commitments, чтобы проверять оба типа операций — арифметические и нетрадиционные — в архитектуре Transformer. Это подходящий фреймворк для выполнения проверяемых и защищенных предсказаний внутри LoRA-файнтюнинга, обеспечивая защиту закрытости входных данных и моделей. GPU-оптимизированная реализация позволяет обрабатывать модели с 13 миллиардами параметров, такими как LLaMA. ## Результаты мы проводим эксперименты на LLaMA и других моделях с 13 миллиардами параметров. Результаты показывают, что zkLoRA позволяет проверять правильность всех операций в LoRA-файнтюнинге, в том числе forward/backward propagation и параметрические обновления. Мы доказываем, что zkLoRA обеспечивает безопасность и конфиденциальность данных, при этом не ухудшая производительность. Это позволяет использовать zkLoRA в сценариях, где необходимо доказать целостность результатов или защитить конфиденциальность моделей. ## Значимость zkLoRA может применяться в сценариях, где требуется проверка целостности результатов, такие как финансовые системы, государственные системы, или решения, выполняемые на небезопасных платформах. Он уменьшает затраты на вычисления, обеспечивает конфиденциальность моделей и позволяет проверить правильность результатов. Это значительно расширяет возможности использования LLMs в различных секторах, где безопасность и проверяемость критически важны. ## Выводы Мы представляем zkLoRA, первую систему, которая способна проверять целостность и защищать LoRA-файнтюнинг в языковых моделях. Мы продемонстрировали, что этот подход эффективен, безопасен и применим к моделям с большим количеством пара