Probabilistic graphical models and Causal AI

Probabilistic graphical models

Probabilistic graphical models are very effective at modelling complex relationships among variables. These might be the relationships between symptoms and diseases, or the relationships between a set of sensor inputs and the state of the system being modelled, or the relationships between cellular metabolic reactions and the genes that encode them, or the relationships between users in a social network about whom we wish to draw inferences. 

Probabilistic graphical models use nodes to represent random variables and graphs to represent joint distributions over variables. By utilising conditional independence, a gigantic joint distribution (over potentially thousands or millions of variables) can be decomposed to local distributions over small subsets of variables, which facilitates efficient inference and learning. 

Projects

  • Probabilistic graphical models for interventional queries

    The project intends to develop methods to suggest how to optimally intervene so that the future state of the system will best suit our interests. The power of probabilistic graphical models to model complex relationships and interactions among a large number of variables facilitates many applications. However, such models only aim to understand the underlying environment. What is ultimately needed in many real-world applications is to suggest how we ought to intervene or act, so as to alter the environment to best suit our interests. The proposed project aims to achieve this using probabilistic graphical models on massive real-world data sets, thus facilitating a variety of applications from health care to commerce and the environment.

    A/Prof Qinfeng Shi; Assistant Professor Julian McAuley; Associate Professor Pawan Mudigonda

  • Compressive sensing based probabilistic graphical models

    Probabilistic Graphical Models (PGMs) use graphs to represent the interactions between random variables and provide a formalism by which to represent complex probabilistic relationships. Despite the success of PGMs in many fields, the learning on real industrial large scale applications is very slow. I will exploit the sparsity and compressibility in PGMs, and turn the large scale PGMs to a number of small scale PGMs. Solving these small scale PGMs and then reversely recover the solutions in the original large scale PGMs in the context of Compressive Sensing. This way, I can effectively deal with large scale PGMs in the computational complexity of small scale PGMs as well as provide theoretical guarantees on the consistency of the solution.

    A/Prof Qinfeng Shi

Causal AI

Causal AI is dedicated to developing the next generation of artificial intelligence — systems that move beyond prediction to understand and influence the true causes behind complex processes.

Our mission is to harness causal reasoning to help and serve humanity, empowering people and organisations to make smarter, earlier, and more sustainable decisions.

The world is changing rapidly. Climate shifts, ecological risks, and social disruptions demand AI that is not only powerful but also responsible, explainable, and resilient. By uncovering root causes, identifying hidden variables, and modelling the outcomes of interventions, we create tools that allow society to go beyond “what is happening” to answer the deeper question:

What is the ideal sequence of actions — given our resources — to achieve the best possible future?

Research Focus

Our group pioneers methods in:

  • Causal Discovery – revealing hidden drivers behind data patterns.
  • Counterfactual Reasoning – answering “What-If” questions with scientific precision.
  • Robust Generalisation – building immunity to spurious correlations and distribution shifts.
  • Intervention Modelling – predicting the impact of real-world actions before they are taken.
  • Optimisation of Outcomes – designing ideal intervention strategies under constraints such as cost, time, or resources.

Causal AI has the power to transform how we make decisions in some of the most complex and high-stakes domains. By moving beyond correlations to identify true causes and simulate interventions, our methods bring clarity, foresight, and resilience where traditional AI often falls short.

Areas of Work 

  • AI for Science & Industry

    • Accelerate the discovery of clean energy materials and sustainable technologies.
    • Optimise industrial operations by identifying true efficiency drivers.
    • Improve manufacturing resilience by modelling intervention impacts on safety, quality, and productivity.
    • Support health and sport science through causal insights into training, performance, and wellbeing.
  • Critical Supply Chains & Infrastructure

    • Uncover the root causes of bottlenecks and disruptions in global supply chains.
    • Model the cascading impacts of shocks (e.g., pandemics, conflicts, extreme weather) on trade and logistics.
    • Test intervention strategies such as diversification of suppliers or rerouting of logistics.
    • Support resilience planning for critical infrastructure such as energy grids, transport networks, and digital systems.
  • Health & Medicine

    • Distinguish causal factors in disease progression, treatment effectiveness, and patient outcomes.
    • Personalise treatment strategies by predicting the impact of medical interventions.
    • Improve public health planning by modelling the effects of policies on populations.
    • Build robust, interpretable models that doctors and policymakers can trust.
  • Agriculture & Food Security

    • Detect crop stress and soil changes before visible damage occurs.
    • Optimise fertiliser, irrigation, and planting strategies for sustainable yield.
    • Model climate-driven risks such as drought, pests, and disease outbreaks.
    • Support food security through resilient farming systems.
  • Water & Environmental Management

    • Reveal hidden drivers of harmful algal blooms and water contamination.
    • Model ecological responses to policy changes or land use interventions.
    • Provide early warnings for floods, fires, and other natural hazards.
    • Support integrated catchment management and resilient water systems.
  • Sustainability & Development

    • Identify leverage points for emissions reduction and biodiversity protection.
    • Explore sustainable development pathways under different policy or investment scenarios.
    • Model land restoration strategies to optimise ecological and economic outcomes.
    • Integrate satellite, climate, and policy data for systems-level planning.
  • Risk, Security & Governance

    • Anticipate systemic risks across finance, environment, and geopolitics.
    • Provide robust evidence for policy and governance under uncertainty.
    • Design interventions that reduce vulnerabilities in critical national systems.
    • Support ethical and responsible AI deployment by exposing hidden biases and unintended consequences.

Our causal AI solutions are already helping:

  • Farmers increase yield while reducing environmental footprint.
  • Agencies anticipate and mitigate ecological risks before crises emerge.
  • Sustainability organisations balance development with stewardship of the Earth.
  • Industries design smarter, safer, and more efficient processes.
  • Governments build resilience into supply chains, infrastructure, and policy decisions.

By focusing on cause-and-effect, we ensure AI is not just reactive, but proactive and truly aligned with long-term human and planetary well-being.

AIML's Causal AI Group is led by Professor Javen Qinfeng Shi, a global leader in Responsible and Causal AI.