Model Description

KETI에서 제공하는 트랜스포머기반 요약 모델인 Ke-T5-small을 기반으로 Finetuning하여 만든 '뉴스 기사 제목 생성 모델'입니다.

학습 데이터는 AIHub에서 제공하는 '문서요약 텍스트 (https://aihub.or.kr/aihubdata/data/view.do?currMenu=115&topMenu=100&aihubDataSe=realm&dataSetSn=97) '의 신문기사들을 사용하였습니다.

원래는 '뉴스 기사 요약 모델'을 만들어 보고 싶었으나, 개인 연구 사정상 컴퓨팅 성능이 좋지 않아 우선 뉴스 제목 생성 모델로 만들어 보았습니다.

실행환경

Window 10

NVIDIA GeForce RTX 3070, 8192 MiB

Python: 3.10.14

PyTorch: 1.12.1

Transformers: 4.46.2

Datasets: 3.2.0

Tokenizers: 0.20.3

차후 뉴스 기사 요약 모델도 만들어보도록 하겠습니다. 감사합니다.