📣 Help Shape the Future of UKRI's Gateway to Research (GtR)

We're improving UKRI's Gateway to Research and are seeking your input! If you would be interested in being interviewed about the improvements we're making and to have your say about how we can make GtR more user-friendly, impactful, and effective for the Research and Innovation community, please email gateway@ukri.org.

FreeML: Engineering Networked Machine Learning via Meta-Free Energy Minimisation

Lead Research Organisation: King's College London
Department Name: Engineering

Abstract

Inspired by neuroscience, informed by information-theoretic principles, and motivated by modern wireless systems architectures integrating artificial intelligence (AI) and communications, this Fellowship sets out to develop a paradigm-shifting framework for networked machine learning (ML) that is centred on the following ideas.

1. Free energy minimisation: According to the free energy principle, agents optimise internal models so as to minimise their information-theoretic surprise vis-a-vis the available data and prior information. This principle offers a basis to reason about epistemic uncertainty ("know when you don't know") in AI agents that is grounded in information-theoretic analyses of out-of-sample generalisation - away from the current narrow focus on point-wise accuracy, towards uncertainty quantification and calibration. A well-calibrated agent can make informed decisions about when to refrain from acting, about when and how to collect or request more data from the environment or other agents, and about how to guard against anomalies or malicious agents.

2. Networked meta-learning: In meta-learning, agents do not share an ML model in full as in conventional, centralised, solutions. Rather, only a meta-model is shared as a means to transfer knowledge across agents, while enabling the optimisation of personalised local models. As advocated by FreeML, meta-models can naturally implement the engineering principle of modularity by encompassing a common repository of functions that can be combined to suit the cognitive needs of each agent. This framework bridges the gap between the dominant centralised or joint learning approaches - including also federated learning - and the individual learning baseline, by means of limited model sharing, while still enabling meaningful cooperation with a controlled privacy loss.

3. Native integration of wireless communication and learning: Conventional wireless systems are based on the principle of separation between computing and communications. In contrast, the native integration of communications and learning advocated by FreeML embeds wireless communication primitives
as part of the data generating and processing model. Like state-of-the-art integrated solutions, the proposed approach aims at fully utilizing radio channel capacity by avoiding inefficiencies due to separate processing. Unlike existing methods, however, the FreeML framework moves away from the standard problem of communicating under uncertainty (on the communication channel) to the novel problem of communicating uncertainty (on the
solution of the cognitive task) under uncertainty (on the communication channel) in order to support networked meta-learning.

Overall, FreeML sets out to study a novel, theoretically principled, paradigm for ML that moves away from the current centralised, accuracy-focused, state of the art in ML to embrace decentralization via wireless connectivity, uncertainty quantification, personalisation, modularity, privacy preservation, and the right to erasure.

FreeML will involve three industrial partners -- Intel, InterDigital, and Samsung AI -- that will provide guidance and feedback on aspects related to implementation efficiency, communications, and integration with wireless networks, respectively.

This Fellowship proposal builds on the PI's unique inter-disciplinary expertise in information theory, ML, and communications, and is intended to enable a step change in the applicant's career towards a leadership position at the intersection of the fields of engineering and ML/AI. Through this programme, the PI will reach out to a diverse community of STEM students, public, regulators, journalists, and academic colleagues across the two fields to advocate for the central role of engineering for reliable and sustainable ML/AI.
 
Description The key results of this work span three major areas: uncertainty-aware AI, networked meta-learning, and the deep integration of wireless communications with machine learning.

1. Smarter AI: Learning with Uncertainty and Confidence
One of the central challenges in AI today is trust-how can an AI system "know what it doesn't know"? Traditionally, machine learning models are trained to make decisions with the highest possible accuracy, but in the real world, uncertainty is inevitable. Simeone's work has leveraged the free energy principle-a concept borrowed from neuroscience-to build AI models that quantify their own uncertainty. This means that instead of blindly making decisions, AI agents can recognize when their knowledge is limited, request more data, or even refuse to act when confidence is too low.

This approach has major implications. In self-driving cars, for instance, an AI agent using these principles wouldn't just classify objects on the road-it would also estimate how reliable its classification is. If it isn't sure whether an object ahead is a pedestrian or a shadow, it could slow down or request more data from other sensors. Similarly, in cybersecurity, AI models trained with these principles can better detect anomalies and adversarial attacks by recognizing when incoming data deviates from what they expect.

So far, the project has introduced novel methodologies based on Bayesian learning and conformal prediction that are being widely considered as viable alternatives to the state of the art to enhance reliability.

2. Networked Meta-Learning: Smarter Collaboration Between AI Agents
Traditionally, machine learning models either rely on a single centralized system (where all data is sent to one location for processing) or use federated learning (where models are trained separately and later combined). Both approaches have limitations-centralized learning raises privacy concerns, while federated learning can struggle with efficiency.

This project introduces a more modular and personalized method. Instead of sharing full models, AI agents exchange meta-models, which contain distilled knowledge rather than raw data. This makes AI systems more adaptable and efficient, allowing them to collaborate while maintaining privacy. For example, in healthcare, hospitals could use this technique to share insights from medical scans without exposing sensitive patient data.

This principle of modularity also allows AI agents to tailor their learning based on their specific needs. A factory robot and a self-driving car may both use machine learning, but they require different skills. Networked meta-learning allows them to build on shared knowledge while maintaining their own specializations. So far, this approach has led to a major paper to be presented at AISTATS that introduces for the first time efficient way to quantify uncertainty in a decentralized fashion.

3. Bridging AI and Wireless Communication: The Future of Connected Intelligence
The final tenet of this project is the seamless integration of machine learning with wireless networks. Traditionally, communications and computing have been treated as separate problems. First, data is transmitted across a network; then, it is processed using AI algorithms. However, in a world increasingly dominated by connected AI systems-such as autonomous drones, smart cities, and real-time translation services-this separation leads to inefficiencies.

This project challenges this divide by treating communication as an integral part of learning itself. This research redefines the problem: rather than just "communicating under uncertainty," his framework addresses how AI systems can communicate their uncertainty while dealing with uncertain wireless conditions. This new perspective allows AI agents to intelligently decide what information to transmit, when to communicate, and how to optimize wireless resources.

In practical terms, this means a self-driving car navigating a busy city won't just transmit massive amounts of raw sensor data. Instead, it will prioritize sending only the most critical and uncertain information to a central server or nearby vehicles, ensuring faster and more efficient decision-making. Similarly, in disaster response scenarios, drone networks could use this principle to relay only the most relevant information to emergency teams, even when communication bandwidth is limited.
Exploitation Route This project is introducing novel methodologies for optimizing and calibrating AI models that are being evaluated by other academic groups around the world.
Sectors Electronics

 
Title GitHub repo for all the papers reported 
Description The repository contains code for all the papers published within my group. 
Type Of Material Computer model/algorithm 
Year Produced 2023 
Provided To Others? Yes  
Impact The code is being used by researchers worldwide. 
URL https://github.com/orgs/kclip/repositories
 
Description NVIDIA 
Organisation NVIDIA
Country Global 
Sector Private 
PI Contribution Development of a new uncertainty-aware ray tracing algorithm for digital twins in wireless systems.
Collaborator Contribution Joint work and provision of open source code
Impact - submitted paper https://arxiv.org/abs/2312.12625
Start Year 2023
 
Description University of Florence 
Organisation University of Florence
Country Italy 
Sector Academic/University 
PI Contribution Development of novel tools for the analysis of quantum statistical learning
Collaborator Contribution Joint research
Impact - Nikoloska I, Simeone O, Banchi L, Velickovic P. Time-warping invariant quantum recurrent neural networks via quantum-classical adaptive gating. Machine Learning: Science and Technology. 2023 Nov 27;4(4):045038. - https://arxiv.org/pdf/2309.11617.pdf
Start Year 2023