🇰🇷 한글 설명
TensorFlow의 TFLite 연산자 중 TransposeConv의 최적화된 구현에서 0으로 나누기 오류가 발생할 수 있는 취약점이 발견되었습니다. 공격자가 stride 값이 0인 모델을 제작할 경우 시스템 오류를 유발할 수 있습니다.
💡 쉬운 설명
특수하게 조작된 인공지능 모델을 실행하면 프로그램이 계산 과정에서 0으로 나누기를 시도하다가 멈춰버릴 수 있습니다. 이는 서비스 중단이나 오류를 일으키는 원인이 됩니다.
🇺🇸 원문 (English)
TensorFlow is an end-to-end open source platform for machine learning. The optimized implementation of the `TransposeConv` TFLite operator is [vulnerable to a division by zero error](https://github.com/tensorflow/tensorflow/blob/0d45ea1ca641b21b73bcf9c00e0179cda284e7e7/tensorflow/lite/kernels/internal/optimized/optimized_ops.h#L5221-L5222). An attacker can craft a model such that `stride_{h,w}` values are 0. Code calling this function must validate these arguments. The fix will be included in TensorFlow 2.5.0. We will also cherrypick this commit on TensorFlow 2.4.2, TensorFlow 2.3.3, TensorFlow 2.2.3 and TensorFlow 2.1.4, as these are also affected and still in supported range.
📍 영향받는 시스템
TensorFlow 2.5.0 미만 버전 (2.4.2, 2.3.3, 2.2.3, 2.1.4 포함)
⚔️ 공격 벡터
stride 값이 0으로 설정된 악의적인 TFLite 모델 파일을 제공하여 실행 유도
🛡️ 대응 방법
TensorFlow를 2.5.0 이상의 최신 버전으로 업데이트하거나 해당 패치가 적용된 버전으로 업그레이드
🔧 검증 코드
출력된 버전 번호가 2.5.0보다 낮은지 확인하여 취약 여부를 판단합니다.
python3 -c 'import tensorflow as tf; print(tf.__version__)'