Augmented Reinforcement Learning Framework For Enhancing Decision-Making In Machine Learning Models Using External Agents

TL;DR

This paper introduces an Augmented Reinforcement Learning framework that integrates external agents, including humans, to improve decision-making, robustness, and accuracy of machine learning models, especially in complex real-world scenarios.

Contribution

The novel ARL framework incorporates two external agents for real-time evaluation and data curation, enhancing reinforcement learning with human and automated oversight.

Findings

Human feedback improves model robustness and accuracy.

ARL achieves higher learning standards in complex environments.

External agents effectively correct decisions and curate data.

Abstract

This work proposes a novel technique Augmented Reinforcement Learning framework for the improvement of decision-making capabilities of machine learning models. The introduction of agents as external overseers checks on model decisions. The external agent can be anyone, like humans or automated scripts, that helps in decision path correction. It seeks to ascertain the priority of the "Garbage-In, Garbage-Out" problem that caused poor data inputs or incorrect actions in reinforcement learning. The ARL framework incorporates two external agents that aid in course correction and the guarantee of quality data at all points of the training cycle. The External Agent 1 is a real-time evaluator, which will provide feedback light of decisions taken by the model, identify suboptimal actions forming the Rejected Data Pipeline. The External Agent 2 helps in selective curation of the provided feedback with relevance and accuracy in business scenarios creates an approved dataset for future training cycles. The validation of the framework is also applied to a real-world scenario, which is "Document Identification and Information Extraction". This problem originates mainly from banking systems, but can be extended anywhere. The method of classification and extraction of information has to be done correctly here. Experimental results show that including human feedback significantly enhances the ability of the model in order to increase robustness and accuracy in making decisions. The augmented approach, with a combination of machine efficiency and human insight, attains a higher learning standard-mainly in complex or ambiguous environments. The findings of this study show that human-in-the-loop reinforcement learning frameworks such as ARL can provide a scalable approach to improving model performance in data-driven applications.