探索贝叶斯网络及其在概率推理中的作用。了解这些图模型如何增强可解释AI,并与 Ultralytics YOLO26 结合使用。
一个贝叶斯网络是一种概率图模型,它使用有向无环图 (DAG) 来表示一组变量及其条件依赖关系。与仅仅将输入映射到输出的“黑箱”算法不同,这些网络明确地建模了不同因素之间的因果关系。这种结构允许数据科学家执行预测建模并在不确定性下进行推理,这使得它们在数据可能不完整或需要将专家领域知识与统计证据相结合的场景中非常有效。
这些网络的核心是贝叶斯定理,一个数学公式,用于随着更多证据或信息的出现而更新假设的概率。在贝叶斯网络中,节点代表变量——例如症状、传感器读数或分类标签——而边(箭头)代表概率依赖关系。如果从节点A到节点B存在链接,则表示A对B有直接影响。这种架构对于可解释人工智能 (XAI)至关重要,因为它允许用户追踪模型的推理路径,提供了复杂深度学习架构通常难以实现的透明度。
这些模型在需要严格风险评估的领域特别相关。通过利用条件概率分布,贝叶斯网络可以回答关于特定变量状态的查询,给定关于其他变量的观测证据。这个过程通常被称为概率推理,与标准神经网络执行的函数逼近不同。
贝叶斯网络广泛部署于决策需要权衡多个不确定因素的行业。
区分贝叶斯网络与其他统计和机器学习模型很重要:
虽然贝叶斯网络处理显式因果图,但现代深度学习模型也输出反映确定性的概率置信度分数。当使用Ultralytics平台等工具在自定义数据集上训练模型时,理解这些概率是解释模型性能的关键。
以下Python代码演示了如何使用预训练模型访问分类任务的概率分布(置信度)。这说明了在现代推理工作流中如何量化确定性。
from ultralytics import YOLO
# Load the YOLO26n-cls classification model
model = YOLO("yolo26n-cls.pt")
# Run inference on an image source
# This returns a results object containing probability data
results = model("https://ultralytics.com/images/bus.jpg")
# Iterate through results to display class probability
for result in results:
# Access the 'probs' attribute for classification probabilities
top_class_index = result.probs.top1
confidence = result.probs.top1conf
print(f"Predicted Class Index: {top_class_index}, Confidence: {confidence:.4f}")

开启您的机器学习未来之旅