
Granite Guardian과 LoRA, 그리고 KV 캐싱으로 정확도와 인프라 비용을 동시에 해결한 여정
글로벌 메신저 LINE에서 발생하는 유해 오픈챗을 차단하기 위한 머신러닝 모니터링 시스템의 고도화 과정을 다룹니다. 기존 모델의 한계를 극복하기 위해 안전성에 특화된 Granite Guardian 소형 디코더를 도입하고, 정교한 데이터 정제와 최적의 추론 엔지니어링을 통해 성능과 비용 효율성을 극대화한 실전 노하우를 담고 있습니다.
실시간으로 생성되는 다량의 텍스트 콘텐츠에서 유해성 판단 시스템을 구축해야 하거나, 소형 LLM(sLLM)을 활용한 가성비 높은 분류 성능 최적화를 적용하고자 하는 엔지니어에게 필독을 권장합니다.
글로벌 메신저인 LINE의 오픈챗 서비스는 하루에도 엄청난 양의 방이 생성 및 수정되어 수동 검수가 불가능한 수준이나, 기존 모니터링 모델은 미세한 가이드라인 변화 및 특정 국가의 판단 기준을 완벽히 만족하지 못했습니다.
수동 검수 데이터의 라벨 불일치를 TF-IDF 아이디어를 기반으로 정제하고, 안전성 과제에 최적화된 소형 2B 디코더 모델인 Granite Guardian 3.1에 LoRA 미세조정(Fine-tuning)을 적용하여 징계 코드와 사유를 동시에 도출하도록 학습시켰습니다.
오프라인 및 온라인 평가 결과 모든 적용 지역에서 정상 및 유해 클래스에 대한 F1 스코어가 대폭 개선되었으며, 임곗값 설정을 통해 엄격한 품질 기준을 만족하며 실시간 자동 검수 적용 범위를 넓혔습니다.
Trade-off
완벽에 가까운 정밀도(Precision)를 확보하기 위해 엄격한 신뢰도 임곗값을 설정함에 따라, 자동으로 완결 처리할 수 있는 비율인 재현율(Recall)이 일부 줄어드는 성능 트레이드오프를 감수해야 했습니다.
IBM Research에서 발표한 안전성 검수 및 위험 요소 탐지(Safety Moderation) 과제에 특화된 소형 언어 모델 제품군입니다.
기존 대형 모델의 가중치를 고정하고 저차원의 행렬 가중치만 추가하여 학습 비용과 메모리 소모를 혁신적으로 줄이는 파인튜닝 기법입니다.
트랜스포머 디코더에서 문장을 생성할 때 이전 시점에 계산된 어텐션 키와 벨류 값을 메모리에 저장하고 재사용하여 연산 속도를 대폭 높여주는 추론 최적화 기법입니다.




