HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

ATHENA: Mathematical Reasoning with Thought Expansion

JB. Kim Hazel Kim Joonghyuk Hahn Yo-Sub Han

ATHENA: Mathematical Reasoning with Thought Expansion

Abstract

Solving math word problems depends on how to articulate the problems, the lens through which models view human linguistic expressions. Real-world settings count on such a method even more due to the diverse practices of the same mathematical operations. Earlier works constrain available thinking processes by limited prediction strategies without considering their significance in acquiring mathematical knowledge. We introduce Attention-based THought Expansion Network Architecture (ATHENA) to tackle the challenges of real-world practices by mimicking human thought expansion mechanisms in the form of neural network propagation. A thought expansion recurrently generates the candidates carrying the thoughts of possible math expressions driven from the previous step and yields reasonable thoughts by selecting the valid pathways to the goal. Our experiments show that ATHENA achieves a new state-of-the-art stage toward the ideal model that is compelling in variant questions even when the informativeness in training examples is restricted.

Code Repositories

the-jb/athena-math
Official
pytorch
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
math-word-problem-solving-on-asdiv-aATHENA (roberta-base)
Execution Accuracy: 86.4
math-word-problem-solving-on-asdiv-aATHENA (roberta-large)
Execution Accuracy: 91
math-word-problem-solving-on-math23kATHENA (roberta-base)
Accuracy (training-test): 84.4
math-word-problem-solving-on-math23kATHENA (roberta-large)
Accuracy (training-test): 86.5
math-word-problem-solving-on-mawpsATHENA (roberta-base)
Accuracy (%): 92.2
math-word-problem-solving-on-mawpsATHENA (roberta-large)
Accuracy (%): 93
math-word-problem-solving-on-svampATHENA (roberta-large)
Execution Accuracy: 54.8
math-word-problem-solving-on-svampATHENA (roberta-base)
Execution Accuracy: 45.6
math-word-problem-solving-on-svamp-1-nATHENA (roberta-large)
Execution Accuracy: 67.8
math-word-problem-solving-on-svamp-1-nATHENA (roberta-base)
Execution Accuracy: 52.5

Build AI with AI

From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp
ATHENA: Mathematical Reasoning with Thought Expansion | Papers | HyperAI