Microsoft выпустила Cognitive Toolkit 2.0, инструмент для машинного обучения

Microsoft Cognitive Toolkit 2.0 теперь общедоступен. Инструмент позволяет создавать, обучать и оценивать собственные нейронные сети, которые затем можно с лёгкостью масштабировать.

Что нового появилось в Cognitive Toolkit 2.0?

Новая версия инструмента находилась в стадии бета-тестирования с октября 2016 года. Исходный код Cognitive Toolkit 2.0 доступен на GitHub. Независимое измерение производительности показало, что на одном графическом процессоре инструмент от Microsoft работает лучше аналогов:

На нескольких графических ускорителях, например, новых Tesla V100 от NVIDIA, улучшение заметно ещё сильнее. Это можно объяснить тесным сотрудничеством двух компаний в этой области.

Cognitive Toolkit 2.0 получил, помимо прочего, три новые функции:

• Поддержка Keras: Keras API был создан для разработки приложений c ИИ. Он предлагает простые интерфейсы, минимизирует количество действий пользователя и обеспечивает четкую и эффективную обратную связь. Поддержка Keras в настоящее время находится в режиме открытого тестирования.
• Поддержка Java и Spark: после обучения модели с использованием Python или BrainScript для оценки модели можно было использовать те же Python и BrainScript или C#. Теперь же для этого можно использовать и новый API для Java. Это позволяет интегрировать модели глубокого обучения в свои приложения на Java и оценивать их на таких платформах, как Spark.
• Сжатие моделей: оценка обучаемой модели на слабых процессорах может уменьшить производительность. В Cognitive Toolkit 2.0 доступны расширения, которые позволяют квантовать реализации нескольких операций. Это повысит производительность моделей Cognitive Toolkit на серверных и встроенных устройствах с небольшой потерей точности оценки.