首页
/ GPyTorch v1.14版本发布:全面支持Python 3.10与PyTorch 2.0

GPyTorch v1.14版本发布:全面支持Python 3.10与PyTorch 2.0

2025-06-14 01:24:13作者:幸俭卉

GPyTorch是基于PyTorch的高斯过程(Gaussian Process)机器学习库,它提供了构建和训练高斯过程模型的灵活框架。高斯过程是一种强大的非参数贝叶斯方法,广泛应用于回归、分类、优化等领域。GPyTorch通过利用PyTorch的自动微分和GPU加速能力,使得高斯过程模型能够高效地处理大规模数据。

核心改进与功能增强

线性核函数维度特异性方差支持

新版本对线性核函数进行了重要扩展,现在支持为每个输入维度设置独立的方差参数。这一改进使得模型能够更好地捕捉不同特征的重要性差异,在特征重要性分析任务中尤为有用。开发者可以通过设置variance_per_dimension=True参数来启用这一功能。

Softmax似然函数权重参数优化

修复了SoftmaxLikelihood中一个长期存在的问题:即使明确设置mixing_weights=False,模型仍会添加可学习的权重参数W。这一修复确保了模型配置的精确性,避免了不必要的参数计算,提高了模型效率。

多维正态分布操作增强

新版本对多维正态分布(MVN)类进行了多项功能增强:

  • 实现了unsqueeze方法,支持在指定维度扩展张量
  • 改进了expand方法,现在支持非惰性(non-lazy)多维正态分布
  • 优化了协方差矩阵的复用机制,减少了不必要的计算

这些改进使得批量处理和多任务学习场景下的操作更加灵活高效。

类型注解与代码质量提升

v1.14版本在代码类型注解方面取得了显著进展:

  • gpytorch.Module基类添加了全面的类型注解
  • 改进了线性核函数的类型提示
  • 开始为ExactGP类添加类型注解

这些改进不仅提升了代码的可读性和可维护性,还为开发者提供了更好的IDE支持,减少了类型相关的错误。

性能优化与错误修复

  • 修复了KeOps核矩阵对角线计算的问题,确保了数值稳定性
  • 改进了梯度计算块的符号处理,提高了优化过程的准确性
  • 优化了幻想(fantasization)后的JIT追踪支持,通过分离new_covar_cache实现了更好的模型序列化

向后兼容性说明

值得注意的是,v1.14版本移除了已被弃用的惰性张量(LazyTensor)实现,开发者需要确保代码中不再使用这些已被移除的功能。同时,该版本全面支持Python 3.10和PyTorch 2.0,建议用户升级到这些较新的版本以获得最佳体验。

应用场景与影响

GPyTorch v1.14的这些改进特别有利于以下应用场景:

  1. 高维特征选择任务:通过维度特异性方差支持,可以更精确地评估各特征的重要性
  2. 多任务学习:增强的多维正态分布操作简化了复杂模型的构建
  3. 生产环境部署:改进的类型系统和JIT支持提高了代码的可靠性

对于高斯过程研究者和实践者来说,这一版本标志着GPyTorch在功能性、稳定性和开发者体验方面的重要进步,为构建更复杂、更高效的高斯过程模型奠定了坚实基础。

登录后查看全文
热门项目推荐

项目优选

收起
kernelkernel
deepin linux kernel
C
22
6
docsdocs
OpenHarmony documentation | OpenHarmony开发者文档
Dockerfile
163
2.05 K
nop-entropynop-entropy
Nop Platform 2.0是基于可逆计算理论实现的采用面向语言编程范式的新一代低代码开发平台,包含基于全新原理从零开始研发的GraphQL引擎、ORM引擎、工作流引擎、报表引擎、规则引擎、批处理引引擎等完整设计。nop-entropy是它的后端部分,采用java语言实现,可选择集成Spring框架或者Quarkus框架。中小企业可以免费商用
Java
8
0
ohos_react_nativeohos_react_native
React Native鸿蒙化仓库
C++
199
279
leetcodeleetcode
🔥LeetCode solutions in any programming language | 多种编程语言实现 LeetCode、《剑指 Offer(第 2 版)》、《程序员面试金典(第 6 版)》题解
Java
60
16
RuoYi-Vue3RuoYi-Vue3
🎉 (RuoYi)官方仓库 基于SpringBoot,Spring Security,JWT,Vue3 & Vite、Element Plus 的前后端分离权限管理系统
Vue
952
558
ShopXO开源商城ShopXO开源商城
🔥🔥🔥ShopXO企业级免费开源商城系统,可视化DIY拖拽装修、包含PC、H5、多端小程序(微信+支付宝+百度+头条&抖音+QQ+快手)、APP、多仓库、多商户、多门店、IM客服、进销存,遵循MIT开源协议发布、基于ThinkPHP8框架研发
JavaScript
96
15
apintoapinto
基于golang开发的网关。具有各种插件,可以自行扩展,即插即用。此外,它可以快速帮助企业管理API服务,提高API服务的稳定性和安全性。
Go
22
0
金融AI编程实战金融AI编程实战
为非计算机科班出身 (例如财经类高校金融学院) 同学量身定制,新手友好,让学生以亲身实践开源开发的方式,学会使用计算机自动化自己的科研/创新工作。案例以量化投资为主线,涉及 Bash、Python、SQL、BI、AI 等全技术栈,培养面向未来的数智化人才 (如数据工程师、数据分析师、数据科学家、数据决策者、量化投资人)。
Python
77
71
giteagitea
喝着茶写代码!最易用的自托管一站式代码托管平台,包含Git托管,代码审查,团队协作,软件包和CI/CD。
Go
17
0