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

AI-Powered CRISPR could Lead to Faster G…

Stanford Medicine researchers have developed an artificial intelligence (AI) tool to help scientists better plan gene-editing experiments. The technology, CRISPR-GPT, acts as a gene-editing “copilot” supported by AI to help...

Groundbreaking AI Aims to Speed Lifesavi…

To solve a problem, we have to see it clearly. Whether it’s an infection by a novel virus or memory-stealing plaques forming in the brains of Alzheimer’s patients, visualizing disease processes...

AI Spots Hidden Signs of Depression in S…

Depression is one of the most common mental health challenges, but its early signs are often overlooked. It is often linked to reduced facial expressivity. However, whether mild depression or...

AI Tools Help Predict Severe Asthma Risk…

Mayo Clinic researchers have developed artificial intelligence (AI) tools that help identify which children with asthma face the highest risk of serious asthma exacerbation and acute respiratory infections. The study...

ChatGPT 4o Therapeutic Chatbot 'Ama…

One of the first randomized controlled trials assessing the effectiveness of a large language model (LLM) chatbot 'Amanda' for relationship support shows that a single session of chatbot therapy...

AI Model Forecasts Disease Risk Decades …

Imagine a future where your medical history could help predict what health conditions you might face in the next two decades. Researchers have developed a generative AI model that uses...

AI Model Indicates Four out of Ten Breas…

A project at Lund University in Sweden has trained an AI model to identify breast cancer patients who could be spared from axillary surgery. The model analyses previously unutilised information...

AI Distinguishes Glioblastoma from Look-…

A Harvard Medical School–led research team has developed an AI tool that can reliably tell apart two look-alike cancers found in the brain but with different origins, behaviors, and treatments. The...

Overcoming the AI Applicability Crisis a…

Opinion Article by Harry Lykostratis, Chief Executive, Open Medical. The government’s 10 Year Health Plan makes a lot of the potential of AI-software to support clinical decision making, improve productivity, and...

Smart Device Uses AI and Bioelectronics …

As a wound heals, it goes through several stages: clotting to stop bleeding, immune system response, scabbing, and scarring. A wearable device called "a-Heal," designed by engineers at the University...

Dartford and Gravesham Implements Clinis…

Dartford and Gravesham NHS Trust has taken a significant step towards a more digital future by rolling out electronic test ordering using Clinisys ICE. The trust deployed the order communications...