0

0

Python使用概率模型处理不确定性预测任务的训练方法【教学】

舞夢輝影

舞夢輝影

发布时间:2025-12-13 19:29:02

|

949人浏览过

|

来源于php中文网

原创

Python概率建模不确定性预测的核心是输出分布而非点估计,需用NLL等概率损失训练、校准评估覆盖率与区间宽度,并注意sigma约束、Dropout开关等工程细节。

python使用概率模型处理不确定性预测任务的训练方法【教学】

Python中用概率模型做不确定性预测,核心不是只输出一个点估计,而是给出预测结果的分布(比如均值+方差、分位数、或完整后验样本),从而量化“有多不确定”。训练的关键在于目标函数要反映概率性——不能只用MSE,得用负对数似然(NLL)、分位数损失、或贝叶斯变分目标等。

选择适配任务的概率建模方式

不同任务对应不同建模思路:

  • 回归带置信区间:用高斯过程(GPyTorch)、或神经网络输出分布参数(如torch.distributions.Normal(loc, scale)),损失用负对数似然
  • 分位数预测(如交通到达时间P10/P50/P90):用分位数损失(Quantile Loss),每个分位数τ单独建模,损失为max(τ×e, (τ−1)×e),其中e是残差
  • 分类+不确定性(如模型是否可信):用MC Dropout、Deep Ensembles 或 Dirichlet网络,训练时最小化ECE相关损失或KL散度
  • 小数据/强先验场景:用PyMC或TensorFlow Probability做全贝叶斯推断,定义先验+似然,用NUTS或VI拟合后验

用PyTorch实现带不确定性输出的神经网络

以单输出回归为例,让网络最后一层输出两个值:mulog_sigma(稳定训练),再构造正态分布:

class UncertainMLP(nn.Module):
    def __init__(self):
        super().__init__()
        self.net = nn.Sequential(nn.Linear(10, 64), nn.ReLU(), nn.Linear(64, 2))
    def forward(self, x):
        out = self.net(x)  # [batch, 2]
        mu, log_sigma = out[:, 0], out[:, 1]
        sigma = torch.exp(log_sigma) + 1e-6
        return torch.distributions.Normal(mu, sigma)

训练时用负对数似然:

Bandy AI
Bandy AI

全球领先的电商设计Agent

下载

立即学习Python免费学习笔记(深入)”;

dist = model(x_batch)
loss = -dist.log_prob(y_batch).mean()  # 核心:最大化似然 → 最小化负对数似然
loss.backward()

评估不确定性质量比评估点精度更关键

光看RMSE不够,要检验预测分布是否“校准”:

  • 预测区间覆盖率(PICP):比如90%置信区间里,真实值实际落在其中的比例是否接近0.9
  • 区间宽度(MPIW):越窄越好,但不能以牺牲覆盖率为代价;可计算平均区间宽度
  • 可靠性图(Reliability Diagram):横轴是预测置信度(如95%),纵轴是实际经验覆盖率,理想是一条y=x线
  • 预期校准误差(ECE):对分类任务常用,把预测置信度分桶后计算平均|准确率−置信度|

避免常见坑

  • 直接用nn.MSELoss训练输出mu, sigma的网络——这会让sigma坍缩到极小值,失去不确定性表达能力
  • 不约束log_sigmasigma下界,导致梯度爆炸或NaN
  • 测试时忘了关闭Dropout或BN,导致不确定性被低估(尤其MC Dropout需多次前向)
  • 用分位数损失时,多个分位数独立训练→可能违反单调性(P10 > P50),可用Monotonic Quantile Regression正则

基本上就这些。概率建模不是加个“分布输出”就完事,关键是训练目标、评估逻辑和工程细节三者对齐。不复杂但容易忽略。

热门AI工具

更多
DeepSeek
DeepSeek

幻方量化公司旗下的开源大模型平台

豆包大模型
豆包大模型

字节跳动自主研发的一系列大型语言模型

通义千问
通义千问

阿里巴巴推出的全能AI助手

腾讯元宝
腾讯元宝

腾讯混元平台推出的AI助手

文心一言
文心一言

文心一言是百度开发的AI聊天机器人,通过对话可以生成各种形式的内容。

讯飞写作
讯飞写作

基于讯飞星火大模型的AI写作工具,可以快速生成新闻稿件、品宣文案、工作总结、心得体会等各种文文稿

即梦AI
即梦AI

一站式AI创作平台,免费AI图片和视频生成。

ChatGPT
ChatGPT

最最强大的AI聊天机器人程序,ChatGPT不单是聊天机器人,还能进行撰写邮件、视频脚本、文案、翻译、代码等任务。

相关专题

更多
Python AI机器学习PyTorch教程_Python怎么用PyTorch和TensorFlow做机器学习
Python AI机器学习PyTorch教程_Python怎么用PyTorch和TensorFlow做机器学习

PyTorch 是一种用于构建深度学习模型的功能完备框架,是一种通常用于图像识别和语言处理等应用程序的机器学习。 使用Python 编写,因此对于大多数机器学习开发者而言,学习和使用起来相对简单。 PyTorch 的独特之处在于,它完全支持GPU,并且使用反向模式自动微分技术,因此可以动态修改计算图形。

24

2025.12.22

Python 深度学习框架与TensorFlow入门
Python 深度学习框架与TensorFlow入门

本专题深入讲解 Python 在深度学习与人工智能领域的应用,包括使用 TensorFlow 搭建神经网络模型、卷积神经网络(CNN)、循环神经网络(RNN)、数据预处理、模型优化与训练技巧。通过实战项目(如图像识别与文本生成),帮助学习者掌握 如何使用 TensorFlow 开发高效的深度学习模型,并将其应用于实际的 AI 问题中。

54

2026.01.07

pytorch是干嘛的
pytorch是干嘛的

pytorch是一个基于python的深度学习框架,提供以下主要功能:动态图计算,提供灵活性。强大的张量操作,实现高效处理。自动微分,简化梯度计算。预构建的神经网络模块,简化模型构建。各种优化器,用于性能优化。想了解更多pytorch的相关内容,可以阅读本专题下面的文章。

433

2024.05.29

Python AI机器学习PyTorch教程_Python怎么用PyTorch和TensorFlow做机器学习
Python AI机器学习PyTorch教程_Python怎么用PyTorch和TensorFlow做机器学习

PyTorch 是一种用于构建深度学习模型的功能完备框架,是一种通常用于图像识别和语言处理等应用程序的机器学习。 使用Python 编写,因此对于大多数机器学习开发者而言,学习和使用起来相对简单。 PyTorch 的独特之处在于,它完全支持GPU,并且使用反向模式自动微分技术,因此可以动态修改计算图形。

24

2025.12.22

java入门学习合集
java入门学习合集

本专题整合了java入门学习指南、初学者项目实战、入门到精通等等内容,阅读专题下面的文章了解更多详细学习方法。

2

2026.01.29

java配置环境变量教程合集
java配置环境变量教程合集

本专题整合了java配置环境变量设置、步骤、安装jdk、避免冲突等等相关内容,阅读专题下面的文章了解更多详细操作。

2

2026.01.29

java成品学习网站推荐大全
java成品学习网站推荐大全

本专题整合了java成品网站、在线成品网站源码、源码入口等等相关内容,阅读专题下面的文章了解更多详细推荐内容。

0

2026.01.29

Java字符串处理使用教程合集
Java字符串处理使用教程合集

本专题整合了Java字符串截取、处理、使用、实战等等教程内容,阅读专题下面的文章了解详细操作教程。

0

2026.01.29

Java空对象相关教程合集
Java空对象相关教程合集

本专题整合了Java空对象相关教程,阅读专题下面的文章了解更多详细内容。

3

2026.01.29

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
最新Python教程 从入门到精通
最新Python教程 从入门到精通

共4课时 | 22.4万人学习

Django 教程
Django 教程

共28课时 | 3.7万人学习

SciPy 教程
SciPy 教程

共10课时 | 1.3万人学习

关于我们 免责申明 举报中心 意见反馈 讲师合作 广告合作 最新更新
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号 技术交流群
PHP中文网订阅号
每天精选资源文章推送

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn | 湘ICP备2023035733号