Применение глубокого обучения с подходом «один против всех» для задачи многоклассовой классификации дефектов металлических поверхностей
2025.
T.13.
№ 4.
id 2032
Сосновская В.Е.
Гаврилова А.Д.
Волкова Е.А.
Котилевец И.Д.
Ильин Д.Ю.
DOI: 10.26102/2310-6018/2025.51.4.025
В статье исследуется задача многоклассовой классификации дефектов металлических поверхностей с использованием методов глубокого обучения. В качестве основного подхода применяется стратегия «один против всех», позволяющая эффективно разделять различные классы дефектов. Для первоначального анализа использовался датасет «NEU», который включал в себя 6 классов повреждений. Полученные оценки качества классификации сравнивались с другими существующими решениями, после чего набор данных был дополнен одним классом из датасета «Severstal: Steel Defect Detection». Предложены две архитектуры сверточных нейронных сетей, каждая из которых была адаптирована под соответствующие классы. Первая архитектура включает в себя 5 сверточных слоев, 5 слоев максимизирующей подвыборки и 2 полносвязных слоя. Вторая содержит на 2 слоя больше: дополнительный сверточный слой и слой максимизирующей подвыборки. Тестирование на данных датасета «NEU» показало высокую эффективность: доля правильных ответов итоговой модели достигает 98,33 %, точность – 98,39 %, полнота – 98,33 % и F1-мера – 98,33 %. Анализ результатов показал, что предложенный подход продемонстрировал эффективность, сравнимую с результатами из других исследований, а предложенные архитектуры не уступают современным решениям. Модель демонстрирует хорошую скорость обработки, до 103 кадров в секунду на центральном процессоре, что делает ее применимой в производстве, позволяя обнаруживать повреждения в реальном времени. После добавления дополнительного класса доля правильных ответов стала равна 97,14 %, точность – 97,24 %, полнота – 97,14 % и F1-мера – 97,12 %, что указывает на устойчивость и масштабируемость предложенного решения на основе подхода «one-vs-all».
Ключевые слова:
нейронные сети, сверточные нейронные сети, датасет, классификация, дефекты металлических поверхностей, глубокое обучение