Алгоритм российской компании N-Tech.Lab для распознавания лиц по фотографиям на базе глубокой нейронной сети и технологий NVIDIA стал лидером мирового чемпионата по распознаванию лиц The MegaFace Benchmark.
The MegaFace Benchmark, мировой чемпионат, организованный Университетом Вашингтона, - это первое соревнование по распознаванию лиц на больших базах фотографий до миллиона изображений. Ранее решения проверялись на базах порядка 10 тыс. фотографий. Решение N-Tech.Lab, российского стартапа, который разрабатывает алгоритмы искусственного интеллекта на базе нейронных сетей, оказалось наиболее эффективным из более 100 решений, участвовавших в чемпионате — точность распознавания составила более 73,3%. Алгоритм Google распознал чуть более 70,5%.
Пример проекции представлений лиц на плоскость. При этом расстояния являются мерой сходства фотографий. На рисунке хорошо заметно, что фотографии одного человека группируются в кластер. Фото также группируются в зависимости от пола, возраста и других атрибутов лица.
Важным преимуществом алгоритма N-Tech.Lab являются невысокие требования к мощностям компьютеров. Для обучения алгоритма компания использовала два ПК на базе 4 графических процессоров NVIDIA GeForce GTX 980 и один ПК на базе 4х карт NVIDIA GeForce GTX Titan Black.
Созданная NVIDIA технология CUDA (Сompute Unified Device Architecture — программно-аппаратная архитектура массивно параллельных вычислений) позволяет решать большую часть вычислительных задач эффективнее, чем при использовании ресурсов CPU. N-Tech.Lab использовала для обучения нейронных сетей NVIDIA cuDNN 3 – актуальную версию GPU-ускоренной библиотеки математических операций для глубоких нейронных сетей, которую разработчики встраивают во фреймворки машинного обучения высокого уровня. Библиотека обеспечивает высокую производительность и низкую нагрузку на память.
N-Tech.Lab намерена совершенствовать созданный компанией алгоритм, в том числе и за счёт расширения сотрудничества с NVIDIA.
«С помощью больших вычислительных систем на базе GPU мы сможем сделать наш алгоритм еще более эффективным и достичь еще большей точности распознавания. Уже сейчас точность нашего алгоритма намного выше точности распознавания человеком. Например, при работе с базами объемом до 10 тысяч фотографий средняя точность распознавания человеком составляет 25%, наш алгоритм дает точность 90%. После того как нейронная сеть обучена, на одном GPU, она способна обрабатывать до 100 фотографий в секунду, этого достаточно для решения большинства практических задач», - отмечает основатель N-tech.Lab Артём Кухаренко.
N-tech.Lab также активно работает над созданием на базе своего алгоритма новых сервисов. Компания готовится запустить несколько коммерческих продуктов с использованием созданного решения. Один из них — сервис знакомств, позволяющий искать людей с внешностью определённого типа. Пользователь сервиса сможет загрузить фотографию понравившегося человека и найти его или похожих на него людей в соцсетях. В настоящий момент ведётся разработка подобного приложения.