Balancing Training Data and Human Knowledge Makes AI Act More Like a Scientist

When you teach a child how to solve puzzles, you can either let them figure it out through trial and error, or you can guide them with some basic rules and tips. Similarly, incorporating rules and tips into AI training - such as the laws of physics - could make them more efficient and more reflective of the real world. However, helping the AI assess the value of different rules can be a tricky task.

Researchers report in the journal Nexus that they have developed a framework for assessing the relative value of rules and data in "informed machine learning models" that incorporate both. They showed that by doing so, they could help the AI incorporate basic laws of the real world and better navigate scientific problems like solving complex mathematical problems and optimizing experimental conditions in chemistry experiments.

"Embedding human knowledge into AI models has the potential to improve their efficiency and ability to make inferences, but the question is how to balance the influence of data and knowledge," says first author Hao Xu of Peking University. "Our framework can be employed to evaluate different knowledge and rules to enhance the predictive capability of deep learning models."

Generative AI models like ChatGPT and Sora are purely data-driven - the models are given training data, and they teach themselves via trial and error. However, with only data to work from, these systems have no way to learn physical laws, such as gravity or fluid dynamics, and they also struggle to perform in situations that differ from their training data. An alternative approach is informed machine learning, in which researchers provide the model with some underlying rules to help guide its training process, but little is known about the relative importance of rules vs data in driving model accuracy.

"We are trying to teach AI models the laws of physics so that they can be more reflective of the real world, which would make them more useful in science and engineering," says senior author Yuntian Chen of the Eastern Institute of Technology, Ningbo.

To improve the performance of informed machine learning, the team developed a framework to calculate the contribution of an individual rule to a given model's predictive accuracy. The researchers also examined interactions between different rules because most informed machine learning models incorporate multiple rules, and having too many rules can cause models to collapse.

This allowed them to optimize models by tweaking the relative influence of different rules and to filter out redundant or interfering rules entirely. They also identified some rules that worked synergistically and other rules that were completely dependent on the presence of other rules.

"We found that the rules have different kinds of relationships, and we use these relationships to make model training faster and get higher accuracy," says Chen.

The researchers say that their framework has broad practical applications in engineering, physics, and chemistry. In the paper, they demonstrated the method’s potential by using it to optimize machine learning models to solve multivariate equations and to predict the results of thin layer chromatography experiments and thereby optimize future experimental chemistry conditions.

Next, the researchers plan to develop their framework into a plugin tool that can be used by AI developers. Ultimately, they also want to train their models so that the models can extract knowledge and rules directly from data, rather than having rules selected by human researchers.

"We want to make it a closed loop by making the model into a real AI scientist," says Chen. "We are working to develop a model that can directly extract knowledge from the data and then use this knowledge to create rules and improve itself."

Hao Xu, Yuntian Chen, Dongxiao Zhang.
Worth of prior knowledge for enhancing deep learning.
Nexus, 2024. doi: 10.1016/j.ynexs.2024.100003

Most Popular Now

Unlocking the 10 Year Health Plan

The government's plan for the NHS is a huge document. Jane Stephenson, chief executive of SPARK TSL, argues the key to unlocking its digital ambitions is to consider what it...

Alcidion Grows Top Talent in the UK, wit…

Alcidion has today announced the addition of three new appointments to their UK-based team, with one internal promotion and two external recruits. Dr Paul Deffley has been announced as the...

AI can Find Cancer Pathologists Miss

Men assessed as healthy after a pathologist analyses their tissue sample may still have an early form of prostate cancer. Using AI, researchers at Uppsala University have been able to...

AI, Full Automation could Expand Artific…

Automated insulin delivery (AID) systems such as the UVA Health-developed artificial pancreas could help more type 1 diabetes patients if the devices become fully automated, according to a new review...

How AI could Speed the Development of RN…

Using artificial intelligence (AI), MIT researchers have come up with a new way to design nanoparticles that can more efficiently deliver RNA vaccines and other types of RNA therapies. After training...

MIT Researchers Use Generative AI to Des…

With help from artificial intelligence, MIT researchers have designed novel antibiotics that can combat two hard-to-treat infections: drug-resistant Neisseria gonorrhoeae and multi-drug-resistant Staphylococcus aureus (MRSA). Using generative AI algorithms, the research...

AI Hybrid Strategy Improves Mammogram In…

A hybrid reading strategy for screening mammography, developed by Dutch researchers and deployed retrospectively to more than 40,000 exams, reduced radiologist workload by 38% without changing recall or cancer detection...

Penn Developed AI Tools and Datasets Hel…

Doctors treating kidney disease have long depended on trial-and-error to find the best therapies for individual patients. Now, new artificial intelligence (AI) tools developed by researchers in the Perelman School...

New Training Year Starts at Siemens Heal…

In September, 197 school graduates will start their vocational training or dual studies in Germany at Siemens Healthineers. 117 apprentices and 80 dual students will begin their careers at Siemens...

Are You Eligible for a Clinical Trial? C…

A new study in the academic journal Machine Learning: Health discovers that ChatGPT can accelerate patient screening for clinical trials, showing promise in reducing delays and improving trial success rates. Researchers...

New AI Tool Addresses Accuracy and Fairn…

A team of researchers at the Icahn School of Medicine at Mount Sinai has developed a new method to identify and reduce biases in datasets used to train machine-learning algorithms...

Global Study Reveals How Patients View M…

How physicians feel about artificial intelligence (AI) in medicine has been studied many times. But what do patients think? A team led by researchers at the Technical University of Munich...