Saved in:
Bibliographic Details
Main Authors: Tong, Yuxuan, Zhang, Xiwen, Wang, Rui, Wu, Ruidong, He, Junxian
Format: Preprint
Published: 2024
Subjects:
Online Access:https://arxiv.org/abs/2407.13690
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866912166964101120
author Tong, Yuxuan
Zhang, Xiwen
Wang, Rui
Wu, Ruidong
He, Junxian
author_facet Tong, Yuxuan
Zhang, Xiwen
Wang, Rui
Wu, Ruidong
He, Junxian
contents Solving mathematical problems requires advanced reasoning abilities and presents notable challenges for large language models. Previous works usually synthesize data from proprietary models to augment existing datasets, followed by instruction tuning to achieve top-tier results. However, our analysis of these datasets reveals severe biases towards easy queries, with frequent failures to generate any correct response for the most challenging queries. Hypothesizing that difficult queries are crucial to learn complex reasoning, we propose Difficulty-Aware Rejection Tuning (DART), a method that allocates difficult queries more trials during the synthesis phase, enabling more extensive training on difficult samples. Utilizing DART, we have created new datasets for mathematical problem-solving that focus more on difficult queries and are substantially smaller than previous ones. Remarkably, our synthesis process solely relies on a 7B-sized open-weight model, without reliance on the commonly used proprietary GPT-4. We fine-tune various base models on our datasets ranging from 7B to 70B in size, resulting in a series of strong models called DART-MATH. In comprehensive in-domain and out-of-domain evaluation on 6 mathematical benchmarks, DART-MATH outperforms vanilla rejection tuning significantly, being superior or comparable to previous arts, despite using much smaller datasets and no proprietary models. Furthermore, our results position our synthetic datasets as the most effective and cost-efficient publicly available resources for advancing mathematical problem-solving.
format Preprint
id arxiv_https___arxiv_org_abs_2407_13690
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle DART-Math: Difficulty-Aware Rejection Tuning for Mathematical Problem-Solving
Tong, Yuxuan
Zhang, Xiwen
Wang, Rui
Wu, Ruidong
He, Junxian
Computation and Language
Artificial Intelligence
Solving mathematical problems requires advanced reasoning abilities and presents notable challenges for large language models. Previous works usually synthesize data from proprietary models to augment existing datasets, followed by instruction tuning to achieve top-tier results. However, our analysis of these datasets reveals severe biases towards easy queries, with frequent failures to generate any correct response for the most challenging queries. Hypothesizing that difficult queries are crucial to learn complex reasoning, we propose Difficulty-Aware Rejection Tuning (DART), a method that allocates difficult queries more trials during the synthesis phase, enabling more extensive training on difficult samples. Utilizing DART, we have created new datasets for mathematical problem-solving that focus more on difficult queries and are substantially smaller than previous ones. Remarkably, our synthesis process solely relies on a 7B-sized open-weight model, without reliance on the commonly used proprietary GPT-4. We fine-tune various base models on our datasets ranging from 7B to 70B in size, resulting in a series of strong models called DART-MATH. In comprehensive in-domain and out-of-domain evaluation on 6 mathematical benchmarks, DART-MATH outperforms vanilla rejection tuning significantly, being superior or comparable to previous arts, despite using much smaller datasets and no proprietary models. Furthermore, our results position our synthetic datasets as the most effective and cost-efficient publicly available resources for advancing mathematical problem-solving.
title DART-Math: Difficulty-Aware Rejection Tuning for Mathematical Problem-Solving
topic Computation and Language
Artificial Intelligence
url https://arxiv.org/abs/2407.13690