🔥 Трансформеры против уязвимостей: как ИИ учится предсказывать последствия атак

Представь ситуацию: у тебя на руках длинный список уязвимостей, каждая из которых может обернуться утечкой данных, потерей доступности или компрометацией доступа. Времени и рук не хватает, чтобы быстро разобрать всё вручную. Было бы здорово, если бы ИИ сам подсказывал, чем именно грозит та или иная уязвимость?

Именно над этим поработали исследователи Bipin Chhetri и Akbar Siami Namin. Они взяли описания уязвимостей из MITRE CWE и попробовали научить модели автоматически определять их последствия. То есть буквально превращать текст про уязвимость в прогноз: что пострадает: конфиденциальность, целостность или доступность.

🔍 Об исследовании

Учёные собрали очищенный датасет из почти 900 записей CWE и выделили пять категорий последствий: Availability, Access Control, Confidentiality, Integrity и «другое». Задача оказалась непростой, т.к. один и тот же баг часто приводит к нескольким эффектам.

Для анализа сравнили разные подходы. С одной стороны — классические нейросети (CNN, LSTM и их комбинации). С другой — современные трансформеры вроде BERT и модель с иерархическим вниманием (HAN).

📊 Результаты экспериментов

BERT буквально разнёс конкурентов: точность предсказаний составила около 97%, а F1-мера — почти 98%. Особенно уверенно он определял уязвимости, влияющие на конфиденциальность.

HAN показал себя хуже — примерно 44% точности. Но у него был плюс: на некоторых классах он ловил нюансы, которые BERT терял.

💡 Где пригодится?

Такой инструмент может автоматически расставлять приоритеты в списке уязвимостей - SOC или баг-трекинг сразу видят, где рискованнее всего.

BERT «чувствует контекст»: различает, когда речь идёт об утечке, а когда — о потере доступности. Это снижает шум в триаже.

Встраивание модели в пайплайн CTI или управление уязвимостями экономит часы рутинного анализа.

⚠️ Ограничения

Датасет небольшой и несбалансированный: например, по метке Integrity качество заметно ниже.
Длинные описания пришлось урезать до 256 токенов — часть информации могла теряться.
И главное — это вспомогательный инструмент. Решения о критичных инцидентах всё равно остаются за человеком.

🔭 Что дальше?

Исследователи предлагают попробовать другие трансформеры (RoBERTa, ALBERT), синтетически расширить датасет и даже объединить BERT с HAN, чтобы сохранить контекст и поймать редкие паттерны.

⚡️ Итог

Современные языковые модели отлично справляются с предсказанием последствий уязвимостей по их описанию. BERT в этом исследовании выдал 97% точности, что делает его мощным кандидатом для автоматизации triage в SOC и системах управления уязвимостями.

🔗 Ссылка на исследование

Stay secure and read SecureTechTalks 📚

#кибербезопасность #vulnerabilities #transformers #BERT #SOC #threatmodeling #cybersecurity
#ИИвбезопасности #uязвимости #SecureTechTalks