Искусственный интеллект «ВКонтакте» научили делать заголовки новостей

Искусственный интеллект сотрудников «ВКонтакте» умеет создавать заголовки новостей на русском и английском языках. Как сообщили в пресс-службе соцсети, для обучения нейросети использовались тексты агентства РИА «Новости» и издания The New York Times.

Нейросеть использует для генерации только текст новости. Алгоритм анализирует его и формулирует заголовок из фрагментов слов с соблюдением падежей и склонений. В работе ИИ используется архитектура Universal Transformer и техника компрессии данных BPE (Byte Pair Encoding).

По словам авторов алгоритма, по некоторым показателям нейросеть показала себя лучше, чем другие существующие на данный момент системы. Однако стандартные метрики не могут корректно оценить случаи, когда автоматические и оригинальные заголовки полностью отличаются, но смысл новости передан верно.

Сейчас разработчики планируют улучшить качество генерируемых заголовков. Подробно алгоритм описан в статье Self-Attentive Model for Headline Generation на сайте библиотеки Корнельского университета.

Иллюстрации: VK

Источник