본문으로 건너뛰기

🔖 북마크 자동분류 프로젝트

개요

북마크 관리와 크롬 익스텐션을 이용하여 쉽게 등록한 북마크가 알아서 특정 카테고리로 분류되게 해보자!

프로젝트 과정

1. 기능 설계

2. Dependencies

RabbitMQ vs Kafka

분산, 대용량, 고성능이 요구되지 않기 때문에 좀 더 가볍고 관리하기 용이한 RabbitMQ 사용.

Model (Bookmark Classification) : ALBERT vs ELECTRA

Replaced Token Detection(RTD)이라는 새로운 pre-training 태스크를 제안하여 모델 크기, 데이터, 컴퓨팅 리소스가 동일한 조건에서 기존 BERT의 성능을 능가하는 ELECTRA 사용.

3. Service Architecture


프로젝트 결과

  • 크롬 익스텐션과 자체 웹사이트를 이용하여 북마크 등록 ( React )
  • Database에 북마크 정보를 저장 ( Spring 기반의 백엔드 서버 )
  • Model 서버에 분류 Request를 전송하여 학습된 모델 서버가 카테고리 분류 결과를 Response 해줌 (RabbitMQ)
  • 사용자의 북마크를 Response 받은 카테고리로 분류