humancompatible.train: Implementing Optimization Algorithms for Stochastically-Constrained Stochastic Optimization Problems

2509.21254v1 cs.LG, math.OC 2025-09-27
Авторы:

Andrii Kliachkin, Jana Lepšová, Gilles Bareilles, Jakub Mareček

Резюме на русском

## Контекст Обучение глубоких нейронных сетей (DNNs) с ограничениями становится все более важной задачей в области глубокого обучения. Эти ограничения могут относиться к различным аспектам, таким как справедливость, безопасность и эффективность, которые требуют соблюдения заданных условий в процессе обучения. Несмотря на интерес к этой области, нет единого стандартного инструмента для такого обучения. Многие разработчики и исследователи сталкиваются с проблемой отсутствия удобных и надежных инструментов для реализации сложных ограничений во время обучения. Данная работа представляет собой попытку заполнить эту нишу, предлагая простой и расширяемый инструмент, который может быть полезен как для академических исследований, так и для использования в промышленности. ## Метод Предложенный пакет `humancompatible.train` основан на PyTorch и представляет собой инструмент для реализации задач обучения DNNs с помощью стохастически ограниченных методов оптимизации. Он поддерживает несколько методов, которые ранее не были реализованы в других пакетах. Разработчики предлагают универсальную архитектуру, которая позволяет легко добавлять новые методы и ограничения. Инструмент также включает в себя модули для мониторинга и оценки результатов обучения с различными метриками, такими как справедливость и точность. Это делает его гибким и привлекательным для различных задач, где необходимо учесть ограничения в процессе обучения. ## Результаты В ходе работы были проведены эксперименты, сравнивающие два метода обучения с стохастическими ограничениями на задаче справедливости в глубоком обучении. Данные для экспериментов были получены из реальных приложений, где справедливость является критическим аспектом. Результаты показали, что один из методов демонстрирует более высокую стабильность и точность в сравнении с другим. Эти результаты были получены с использованием описанной модели `humancompatible.train`, что демонстрирует её эффективность в реализации и сравнении различных методов для задач с ограничениями. ## Значимость Инструмент `humancompatible.train` имеет широкое применение в различных областях, где требуется учитывать ограничения в процессе обучения DNNs. Например, он может быть использован в сферах финансов, здравоохранения, справедливости в рекомендательных системах. Одним из основных преимуществ является его простота и расширяемость, что позволяет легко адаптировать его под конкретные задачи. Данный подход может способствовать развитию лучших практик в обучении DNNs с ограничениями, а также способствовать улучшению безопасности и справедливости в системах, основанных на глубоком обучении. ## Выводы Работа п

Abstract

There has been a considerable interest in constrained training of deep neural networks (DNNs) recently for applications such as fairness and safety. Several toolkits have been proposed for this task, yet there is still no industry standard. We present humancompatible.train (https://github.com/humancompatible/train), an easily-extendable PyTorch-based Python package for training DNNs with stochastic constraints. We implement multiple previously unimplemented algorithms for stochastically constrained stochastic optimization. We demonstrate the toolkit use by comparing two algorithms on a deep learning task with fairness constraints.

Ссылки и действия