기학습된 언어 모델을 활용하여 자연어를 이해할 수 있는 한글 수학 서술형 문제 풀이 모델
본 연구는 인공지능산업원천기술개발 사업으로 2021.07.01.~2021.12.31.(6개월) 동안 진행된 과제입니다.
본 연구는 한글 수학 문장형 문제(MWP, Math Word Problem)를 푸는 심층신경망 기반 모델을 개발합니다.
수학 문장형 문제는 자연어로 구성된 서술형 수학 문제에 대해 풀이과정과 해답을 제시하는 문제를 의미합니다.
다음과 같은 코드로 실행할 수 있습니다.
python main.py --use_isc True --beam_size 1 --batch_size 32 --seed 951206 --gpu 0 --dataset 'Korean_MWPs'
data 폴더 안에 모델을 학습 및 평가시키고자 하는 데이터를 넣어주어야 합니다.
한국어 수학 문장형 문제 데이터를 이곳에서 찾으실 수 있습니다.
위의 '실행 방법'의 코드를 실행하면 다음과 같은 결과를 얻을 수 있습니다.
test_A accuracy | test_B accuracy | validation_accuracy |
---|---|---|
92.5% | 87.7% | 87.6% |