首页 > 最新文献

Proceedings of the ... International Conference on Machine Learning. International Conference on Machine Learning最新文献

英文 中文
Provably Efficient Representation Learning with Tractable Planning in Low-Rank POMDP 低秩POMDP中具有可处理计划的可证明高效表示学习
Jiacheng Guo, Zihao Li, Huazheng Wang, Mengdi Wang, Zhuoran Yang, Xuezhou Zhang
In this paper, we study representation learning in partially observable Markov Decision Processes (POMDPs), where the agent learns a decoder function that maps a series of high-dimensional raw observations to a compact representation and uses it for more efficient exploration and planning. We focus our attention on the sub-classes of textit{$gamma$-observable} and textit{decodable POMDPs}, for which it has been shown that statistically tractable learning is possible, but there has not been any computationally efficient algorithm. We first present an algorithm for decodable POMDPs that combines maximum likelihood estimation (MLE) and optimism in the face of uncertainty (OFU) to perform representation learning and achieve efficient sample complexity, while only calling supervised learning computational oracles. We then show how to adapt this algorithm to also work in the broader class of $gamma$-observable POMDPs.
在本文中,我们研究了部分可观察马尔可夫决策过程(pomdp)中的表示学习,其中智能体学习了一个解码器函数,该函数将一系列高维原始观察映射到紧凑的表示,并使用它进行更有效的探索和规划。我们将注意力集中在textit{$gamma$-可观察的}和textit{可解码的pomdp}的子类上,已经证明统计上可处理的学习是可能的,但还没有任何计算效率高的算法。我们首先提出了一种可解码pomdp的算法,该算法结合了面对不确定性(OFU)的最大似然估计(MLE)和乐观主义来执行表示学习并实现有效的样本复杂度,同时只调用监督学习计算预言。然后,我们将展示如何使该算法也适用于更广泛的$gamma$ -可观察pomdp类。
{"title":"Provably Efficient Representation Learning with Tractable Planning in Low-Rank POMDP","authors":"Jiacheng Guo, Zihao Li, Huazheng Wang, Mengdi Wang, Zhuoran Yang, Xuezhou Zhang","doi":"10.48550/arXiv.2306.12356","DOIUrl":"https://doi.org/10.48550/arXiv.2306.12356","url":null,"abstract":"In this paper, we study representation learning in partially observable Markov Decision Processes (POMDPs), where the agent learns a decoder function that maps a series of high-dimensional raw observations to a compact representation and uses it for more efficient exploration and planning. We focus our attention on the sub-classes of textit{$gamma$-observable} and textit{decodable POMDPs}, for which it has been shown that statistically tractable learning is possible, but there has not been any computationally efficient algorithm. We first present an algorithm for decodable POMDPs that combines maximum likelihood estimation (MLE) and optimism in the face of uncertainty (OFU) to perform representation learning and achieve efficient sample complexity, while only calling supervised learning computational oracles. We then show how to adapt this algorithm to also work in the broader class of $gamma$-observable POMDPs.","PeriodicalId":74529,"journal":{"name":"Proceedings of the ... International Conference on Machine Learning. International Conference on Machine Learning","volume":"18 1","pages":"11967-11997"},"PeriodicalIF":0.0,"publicationDate":"2023-06-21","publicationTypes":"Journal Article","fieldsOfStudy":null,"isOpenAccess":false,"openAccessPdf":"","citationCount":null,"resultStr":null,"platform":"Semanticscholar","paperid":"78245846","PeriodicalName":null,"FirstCategoryId":null,"ListUrlMain":null,"RegionNum":0,"RegionCategory":"","ArticlePicture":[],"TitleCN":null,"AbstractTextCN":null,"PMCID":"","EPubDate":null,"PubModel":null,"JCR":null,"JCRName":null,"Score":null,"Total":0}
引用次数: 1
Data Structures for Density Estimation 密度估计的数据结构
Anders Aamand, Alexandr Andoni, Justin Y. Chen, P. Indyk, Shyam Narayanan, Sandeep Silwal
We study statistical/computational tradeoffs for the following density estimation problem: given $k$ distributions $v_1, ldots, v_k$ over a discrete domain of size $n$, and sampling access to a distribution $p$, identify $v_i$ that is"close"to $p$. Our main result is the first data structure that, given a sublinear (in $n$) number of samples from $p$, identifies $v_i$ in time sublinear in $k$. We also give an improved version of the algorithm of Acharya et al. (2018) that reports $v_i$ in time linear in $k$. The experimental evaluation of the latter algorithm shows that it achieves a significant reduction in the number of operations needed to achieve a given accuracy compared to prior work.
我们研究了以下密度估计问题的统计/计算权衡:给定$k$分布$v_1, ldots, v_k$在大小为$n$的离散域上,以及对分布$p$的抽样访问,确定$v_i$“接近”$p$。我们的主要结果是第一个数据结构,给定来自$p$的次线性(在$n$中)样本数量,识别$k$中的次线性时间$v_i$。我们还给出了Acharya等人(2018)算法的改进版本,该算法在$k$中报告$v_i$的时间线性。后一种算法的实验评估表明,与之前的工作相比,它实现了实现给定精度所需的操作次数的显着减少。
{"title":"Data Structures for Density Estimation","authors":"Anders Aamand, Alexandr Andoni, Justin Y. Chen, P. Indyk, Shyam Narayanan, Sandeep Silwal","doi":"10.48550/arXiv.2306.11312","DOIUrl":"https://doi.org/10.48550/arXiv.2306.11312","url":null,"abstract":"We study statistical/computational tradeoffs for the following density estimation problem: given $k$ distributions $v_1, ldots, v_k$ over a discrete domain of size $n$, and sampling access to a distribution $p$, identify $v_i$ that is\"close\"to $p$. Our main result is the first data structure that, given a sublinear (in $n$) number of samples from $p$, identifies $v_i$ in time sublinear in $k$. We also give an improved version of the algorithm of Acharya et al. (2018) that reports $v_i$ in time linear in $k$. The experimental evaluation of the latter algorithm shows that it achieves a significant reduction in the number of operations needed to achieve a given accuracy compared to prior work.","PeriodicalId":74529,"journal":{"name":"Proceedings of the ... International Conference on Machine Learning. International Conference on Machine Learning","volume":"56 1","pages":"1-18"},"PeriodicalIF":0.0,"publicationDate":"2023-06-20","publicationTypes":"Journal Article","fieldsOfStudy":null,"isOpenAccess":false,"openAccessPdf":"","citationCount":null,"resultStr":null,"platform":"Semanticscholar","paperid":"88629894","PeriodicalName":null,"FirstCategoryId":null,"ListUrlMain":null,"RegionNum":0,"RegionCategory":"","ArticlePicture":[],"TitleCN":null,"AbstractTextCN":null,"PMCID":"","EPubDate":null,"PubModel":null,"JCR":null,"JCRName":null,"Score":null,"Total":0}
引用次数: 1
A Universal Unbiased Method for Classification from Aggregate Observations 一种普遍无偏的综合观测分类方法
Zixi Wei, Lei Feng, Bo Han, Tongliang Liu, Gang Niu, Xiaofeng Zhu, H. Shen
In conventional supervised classification, true labels are required for individual instances. However, it could be prohibitive to collect the true labels for individual instances, due to privacy concerns or unaffordable annotation costs. This motivates the study on classification from aggregate observations (CFAO), where the supervision is provided to groups of instances, instead of individual instances. CFAO is a generalized learning framework that contains various learning problems, such as multiple-instance learning and learning from label proportions. The goal of this paper is to present a novel universal method of CFAO, which holds an unbiased estimator of the classification risk for arbitrary losses -- previous research failed to achieve this goal. Practically, our method works by weighing the importance of each label for each instance in the group, which provides purified supervision for the classifier to learn. Theoretically, our proposed method not only guarantees the risk consistency due to the unbiased risk estimator but also can be compatible with arbitrary losses. Extensive experiments on various problems of CFAO demonstrate the superiority of our proposed method.
在传统的监督分类中,单个实例需要真实标签。然而,由于隐私问题或无法负担的注释成本,收集单个实例的真实标签可能是令人望而却步的。这激发了从总体观察(CFAO)中进行分类的研究,其中监督提供给实例组,而不是单个实例。CFAO是一个广义的学习框架,它包含多种学习问题,如多实例学习和标签比例学习。本文的目标是提出一种新的通用的CFAO方法,该方法具有任意损失分类风险的无偏估计量,以往的研究未能实现这一目标。实际上,我们的方法通过权衡组中每个实例的每个标签的重要性来工作,这为分类器的学习提供了纯粹的监督。从理论上讲,由于风险估计量无偏,我们提出的方法不仅保证了风险一致性,而且可以兼容任意损失。在各种CFAO问题上的大量实验证明了我们提出的方法的优越性。
{"title":"A Universal Unbiased Method for Classification from Aggregate Observations","authors":"Zixi Wei, Lei Feng, Bo Han, Tongliang Liu, Gang Niu, Xiaofeng Zhu, H. Shen","doi":"10.48550/arXiv.2306.11343","DOIUrl":"https://doi.org/10.48550/arXiv.2306.11343","url":null,"abstract":"In conventional supervised classification, true labels are required for individual instances. However, it could be prohibitive to collect the true labels for individual instances, due to privacy concerns or unaffordable annotation costs. This motivates the study on classification from aggregate observations (CFAO), where the supervision is provided to groups of instances, instead of individual instances. CFAO is a generalized learning framework that contains various learning problems, such as multiple-instance learning and learning from label proportions. The goal of this paper is to present a novel universal method of CFAO, which holds an unbiased estimator of the classification risk for arbitrary losses -- previous research failed to achieve this goal. Practically, our method works by weighing the importance of each label for each instance in the group, which provides purified supervision for the classifier to learn. Theoretically, our proposed method not only guarantees the risk consistency due to the unbiased risk estimator but also can be compatible with arbitrary losses. Extensive experiments on various problems of CFAO demonstrate the superiority of our proposed method.","PeriodicalId":74529,"journal":{"name":"Proceedings of the ... International Conference on Machine Learning. International Conference on Machine Learning","volume":"17 1","pages":"36804-36820"},"PeriodicalIF":0.0,"publicationDate":"2023-06-20","publicationTypes":"Journal Article","fieldsOfStudy":null,"isOpenAccess":false,"openAccessPdf":"","citationCount":null,"resultStr":null,"platform":"Semanticscholar","paperid":"86884888","PeriodicalName":null,"FirstCategoryId":null,"ListUrlMain":null,"RegionNum":0,"RegionCategory":"","ArticlePicture":[],"TitleCN":null,"AbstractTextCN":null,"PMCID":"","EPubDate":null,"PubModel":null,"JCR":null,"JCRName":null,"Score":null,"Total":0}
引用次数: 1
LoSparse: Structured Compression of Large Language Models based on Low-Rank and Sparse Approximation LoSparse:基于低秩和稀疏逼近的大型语言模型的结构化压缩
Yixiao Li, Yifan Yu, Qingru Zhang, Chen Liang, Pengcheng He, Weizhu Chen, Tuo Zhao
Transformer models have achieved remarkable results in various natural language tasks, but they are often prohibitively large, requiring massive memories and computational resources. To reduce the size and complexity of these models, we propose LoSparse (Low-Rank and Sparse approximation), a novel model compression technique that approximates a weight matrix by the sum of a low-rank matrix and a sparse matrix. Our method combines the advantages of both low-rank approximations and pruning, while avoiding their limitations. Low-rank approximation compresses the coherent and expressive parts in neurons, while pruning removes the incoherent and non-expressive parts in neurons. Pruning enhances the diversity of low-rank approximations, and low-rank approximation prevents pruning from losing too many expressive neurons. We evaluate our method on natural language understanding, question answering, and natural language generation tasks. We show that it significantly outperforms existing compression methods.
Transformer模型在各种自然语言任务中取得了显著的成果,但是它们通常过于庞大,需要大量的内存和计算资源。为了减少这些模型的大小和复杂性,我们提出了LoSparse (Low-Rank and Sparse approximation),这是一种新的模型压缩技术,它通过一个低秩矩阵和一个稀疏矩阵的和来逼近一个权重矩阵。我们的方法结合了低秩近似和剪枝的优点,同时避免了它们的局限性。低秩逼近压缩神经元中的相干和表达部分,而剪枝则去除神经元中的不相干和非表达部分。剪枝增强了低秩近似的多样性,低秩近似可以防止剪枝损失太多的表达神经元。我们在自然语言理解、问题回答和自然语言生成任务上评估了我们的方法。我们表明,它明显优于现有的压缩方法。
{"title":"LoSparse: Structured Compression of Large Language Models based on Low-Rank and Sparse Approximation","authors":"Yixiao Li, Yifan Yu, Qingru Zhang, Chen Liang, Pengcheng He, Weizhu Chen, Tuo Zhao","doi":"10.48550/arXiv.2306.11222","DOIUrl":"https://doi.org/10.48550/arXiv.2306.11222","url":null,"abstract":"Transformer models have achieved remarkable results in various natural language tasks, but they are often prohibitively large, requiring massive memories and computational resources. To reduce the size and complexity of these models, we propose LoSparse (Low-Rank and Sparse approximation), a novel model compression technique that approximates a weight matrix by the sum of a low-rank matrix and a sparse matrix. Our method combines the advantages of both low-rank approximations and pruning, while avoiding their limitations. Low-rank approximation compresses the coherent and expressive parts in neurons, while pruning removes the incoherent and non-expressive parts in neurons. Pruning enhances the diversity of low-rank approximations, and low-rank approximation prevents pruning from losing too many expressive neurons. We evaluate our method on natural language understanding, question answering, and natural language generation tasks. We show that it significantly outperforms existing compression methods.","PeriodicalId":74529,"journal":{"name":"Proceedings of the ... International Conference on Machine Learning. International Conference on Machine Learning","volume":"62 1","pages":"20336-20350"},"PeriodicalIF":0.0,"publicationDate":"2023-06-20","publicationTypes":"Journal Article","fieldsOfStudy":null,"isOpenAccess":false,"openAccessPdf":"","citationCount":null,"resultStr":null,"platform":"Semanticscholar","paperid":"79702309","PeriodicalName":null,"FirstCategoryId":null,"ListUrlMain":null,"RegionNum":0,"RegionCategory":"","ArticlePicture":[],"TitleCN":null,"AbstractTextCN":null,"PMCID":"","EPubDate":null,"PubModel":null,"JCR":null,"JCRName":null,"Score":null,"Total":0}
引用次数: 6
Stable and Consistent Prediction of 3D Characteristic Orientation via Invariant Residual Learning 基于不变残差学习的三维特征方向稳定一致预测
Seungwook Kim, Chunghyun Park, Yoonwoo Jeong, Jaesik Park, Minsu Cho
Learning to predict reliable characteristic orientations of 3D point clouds is an important yet challenging problem, as different point clouds of the same class may have largely varying appearances. In this work, we introduce a novel method to decouple the shape geometry and semantics of the input point cloud to achieve both stability and consistency. The proposed method integrates shape-geometry-based SO(3)-equivariant learning and shape-semantics-based SO(3)-invariant residual learning, where a final characteristic orientation is obtained by calibrating an SO(3)-equivariant orientation hypothesis using an SO(3)-invariant residual rotation. In experiments, the proposed method not only demonstrates superior stability and consistency but also exhibits state-of-the-art performances when applied to point cloud part segmentation, given randomly rotated inputs.
学习预测三维点云的可靠特征方向是一个重要而又具有挑战性的问题,因为同一类别的不同点云可能具有很大差异的外观。在这项工作中,我们引入了一种新的方法来解耦输入点云的形状几何和语义,以实现稳定性和一致性。该方法将基于形状几何的SO(3)等变学习和基于形状语义的SO(3)不变残差学习相结合,通过使用SO(3)不变残差旋转校准SO(3)等变方向假设来获得最终特征方向。在实验中,该方法不仅表现出优异的稳定性和一致性,而且在随机旋转输入的点云部分分割中也表现出最先进的性能。
{"title":"Stable and Consistent Prediction of 3D Characteristic Orientation via Invariant Residual Learning","authors":"Seungwook Kim, Chunghyun Park, Yoonwoo Jeong, Jaesik Park, Minsu Cho","doi":"10.48550/arXiv.2306.11406","DOIUrl":"https://doi.org/10.48550/arXiv.2306.11406","url":null,"abstract":"Learning to predict reliable characteristic orientations of 3D point clouds is an important yet challenging problem, as different point clouds of the same class may have largely varying appearances. In this work, we introduce a novel method to decouple the shape geometry and semantics of the input point cloud to achieve both stability and consistency. The proposed method integrates shape-geometry-based SO(3)-equivariant learning and shape-semantics-based SO(3)-invariant residual learning, where a final characteristic orientation is obtained by calibrating an SO(3)-equivariant orientation hypothesis using an SO(3)-invariant residual rotation. In experiments, the proposed method not only demonstrates superior stability and consistency but also exhibits state-of-the-art performances when applied to point cloud part segmentation, given randomly rotated inputs.","PeriodicalId":74529,"journal":{"name":"Proceedings of the ... International Conference on Machine Learning. International Conference on Machine Learning","volume":"26 1","pages":"16787-16806"},"PeriodicalIF":0.0,"publicationDate":"2023-06-20","publicationTypes":"Journal Article","fieldsOfStudy":null,"isOpenAccess":false,"openAccessPdf":"","citationCount":null,"resultStr":null,"platform":"Semanticscholar","paperid":"85731520","PeriodicalName":null,"FirstCategoryId":null,"ListUrlMain":null,"RegionNum":0,"RegionCategory":"","ArticlePicture":[],"TitleCN":null,"AbstractTextCN":null,"PMCID":"","EPubDate":null,"PubModel":null,"JCR":null,"JCRName":null,"Score":null,"Total":0}
引用次数: 0
BNN-DP: Robustness Certification of Bayesian Neural Networks via Dynamic Programming 基于动态规划的贝叶斯神经网络鲁棒性证明
Steven Adams, A. Patané, Morteza Lahijanian, L. Laurenti
In this paper, we introduce BNN-DP, an efficient algorithmic framework for analysis of adversarial robustness of Bayesian Neural Networks (BNNs). Given a compact set of input points $Tsubset mathbb{R}^n$, BNN-DP computes lower and upper bounds on the BNN's predictions for all the points in $T$. The framework is based on an interpretation of BNNs as stochastic dynamical systems, which enables the use of Dynamic Programming (DP) algorithms to bound the prediction range along the layers of the network. Specifically, the method uses bound propagation techniques and convex relaxations to derive a backward recursion procedure to over-approximate the prediction range of the BNN with piecewise affine functions. The algorithm is general and can handle both regression and classification tasks. On a set of experiments on various regression and classification tasks and BNN architectures, we show that BNN-DP outperforms state-of-the-art methods by up to four orders of magnitude in both tightness of the bounds and computational efficiency.
本文介绍了一种分析贝叶斯神经网络(BNNs)对抗鲁棒性的有效算法框架BNN-DP。给定一个紧凑的输入点集$T子集mathbb{R}^n$, BNN- dp计算BNN对$T$中所有点的预测的下界和上界。该框架基于对bnn作为随机动态系统的解释,这使得使用动态规划(DP)算法可以沿网络层绑定预测范围。具体来说,该方法利用界传播技术和凸松弛导出了一个反向递归过程,以分段仿射函数过度逼近BNN的预测范围。该算法具有通用性,可以同时处理回归和分类任务。在一系列关于各种回归和分类任务以及BNN架构的实验中,我们表明BNN- dp在边界的紧密性和计算效率方面都优于最先进的方法多达四个数量级。
{"title":"BNN-DP: Robustness Certification of Bayesian Neural Networks via Dynamic Programming","authors":"Steven Adams, A. Patané, Morteza Lahijanian, L. Laurenti","doi":"10.48550/arXiv.2306.10742","DOIUrl":"https://doi.org/10.48550/arXiv.2306.10742","url":null,"abstract":"In this paper, we introduce BNN-DP, an efficient algorithmic framework for analysis of adversarial robustness of Bayesian Neural Networks (BNNs). Given a compact set of input points $Tsubset mathbb{R}^n$, BNN-DP computes lower and upper bounds on the BNN's predictions for all the points in $T$. The framework is based on an interpretation of BNNs as stochastic dynamical systems, which enables the use of Dynamic Programming (DP) algorithms to bound the prediction range along the layers of the network. Specifically, the method uses bound propagation techniques and convex relaxations to derive a backward recursion procedure to over-approximate the prediction range of the BNN with piecewise affine functions. The algorithm is general and can handle both regression and classification tasks. On a set of experiments on various regression and classification tasks and BNN architectures, we show that BNN-DP outperforms state-of-the-art methods by up to four orders of magnitude in both tightness of the bounds and computational efficiency.","PeriodicalId":74529,"journal":{"name":"Proceedings of the ... International Conference on Machine Learning. International Conference on Machine Learning","volume":"39 1","pages":"133-151"},"PeriodicalIF":0.0,"publicationDate":"2023-06-19","publicationTypes":"Journal Article","fieldsOfStudy":null,"isOpenAccess":false,"openAccessPdf":"","citationCount":null,"resultStr":null,"platform":"Semanticscholar","paperid":"90603034","PeriodicalName":null,"FirstCategoryId":null,"ListUrlMain":null,"RegionNum":0,"RegionCategory":"","ArticlePicture":[],"TitleCN":null,"AbstractTextCN":null,"PMCID":"","EPubDate":null,"PubModel":null,"JCR":null,"JCRName":null,"Score":null,"Total":0}
引用次数: 0
Stabilizing GANs' Training with Brownian Motion Controller 基于布朗运动控制器的稳定GANs训练
Tianjiao Luo, Ziyu Zhu, Jianfei Chen, Jun Zhu
The training process of generative adversarial networks (GANs) is unstable and does not converge globally. In this paper, we examine the stability of GANs from the perspective of control theory and propose a universal higher-order noise-based controller called Brownian Motion Controller (BMC). Starting with the prototypical case of Dirac-GANs, we design a BMC to retrieve precisely the same but reachable optimal equilibrium. We theoretically prove that the training process of DiracGANs-BMC is globally exponential stable and derive bounds on the rate of convergence. Then we extend our BMC to normal GANs and provide implementation instructions on GANs-BMC. Our experiments show that our GANs-BMC effectively stabilizes GANs' training under StyleGANv2-ada frameworks with a faster rate of convergence, a smaller range of oscillation, and better performance in terms of FID score.
生成式对抗网络(GANs)的训练过程是不稳定的,不具有全局收敛性。本文从控制论的角度研究了gan的稳定性,提出了一种通用的基于噪声的高阶控制器布朗运动控制器(BMC)。从dirac - gan的原型案例开始,我们设计了一个BMC来检索完全相同但可达到的最佳平衡。从理论上证明了diracgass - bmc的训练过程是全局指数稳定的,并给出了收敛速度的界。然后,我们将BMC扩展到普通gan上,并提供了gan -BMC的实现说明。我们的实验表明,我们的GANs- bmc有效地稳定了StyleGANv2-ada框架下的GANs训练,具有更快的收敛速度,更小的振荡范围,并且在FID评分方面具有更好的性能。
{"title":"Stabilizing GANs' Training with Brownian Motion Controller","authors":"Tianjiao Luo, Ziyu Zhu, Jianfei Chen, Jun Zhu","doi":"10.48550/arXiv.2306.10468","DOIUrl":"https://doi.org/10.48550/arXiv.2306.10468","url":null,"abstract":"The training process of generative adversarial networks (GANs) is unstable and does not converge globally. In this paper, we examine the stability of GANs from the perspective of control theory and propose a universal higher-order noise-based controller called Brownian Motion Controller (BMC). Starting with the prototypical case of Dirac-GANs, we design a BMC to retrieve precisely the same but reachable optimal equilibrium. We theoretically prove that the training process of DiracGANs-BMC is globally exponential stable and derive bounds on the rate of convergence. Then we extend our BMC to normal GANs and provide implementation instructions on GANs-BMC. Our experiments show that our GANs-BMC effectively stabilizes GANs' training under StyleGANv2-ada frameworks with a faster rate of convergence, a smaller range of oscillation, and better performance in terms of FID score.","PeriodicalId":74529,"journal":{"name":"Proceedings of the ... International Conference on Machine Learning. International Conference on Machine Learning","volume":"23 1","pages":"23140-23156"},"PeriodicalIF":0.0,"publicationDate":"2023-06-18","publicationTypes":"Journal Article","fieldsOfStudy":null,"isOpenAccess":false,"openAccessPdf":"","citationCount":null,"resultStr":null,"platform":"Semanticscholar","paperid":"86945652","PeriodicalName":null,"FirstCategoryId":null,"ListUrlMain":null,"RegionNum":0,"RegionCategory":"","ArticlePicture":[],"TitleCN":null,"AbstractTextCN":null,"PMCID":"","EPubDate":null,"PubModel":null,"JCR":null,"JCRName":null,"Score":null,"Total":0}
引用次数: 0
Graph Ladling: Shockingly Simple Parallel GNN Training without Intermediate Communication 图Ladling:令人震惊的简单并行GNN训练没有中间通信
A. Jaiswal, Shiwei Liu, Tianlong Chen, Ying Ding, Zhangyang Wang
Graphs are omnipresent and GNNs are a powerful family of neural networks for learning over graphs. Despite their popularity, scaling GNNs either by deepening or widening suffers from prevalent issues of unhealthy gradients, over-smoothening, information squashing, which often lead to sub-standard performance. In this work, we are interested in exploring a principled way to scale GNNs capacity without deepening or widening, which can improve its performance across multiple small and large graphs. Motivated by the recent intriguing phenomenon of model soups, which suggest that fine-tuned weights of multiple large-language pre-trained models can be merged to a better minima, we argue to exploit the fundamentals of model soups to mitigate the aforementioned issues of memory bottleneck and trainability during GNNs scaling. More specifically, we propose not to deepen or widen current GNNs, but instead present a data-centric perspective of model soups tailored for GNNs, i.e., to build powerful GNNs. By dividing giant graph data, we build multiple independently and parallelly trained weaker GNNs (soup ingredient) without any intermediate communication, and combine their strength using a greedy interpolation soup procedure to achieve state-of-the-art performance. Compared to concurrent distributed GNN training works such as Jiong et. al. 2023, we train each soup ingredient by sampling different subgraphs per epoch and their respective sub-models are merged only after being fully trained (rather than intermediately so). Moreover, we provide a wide variety of model soup preparation techniques by leveraging state-of-the-art graph sampling and graph partitioning approaches that can handle large graphs. Codes are available at: url{https://github.com/VITA-Group/graph_ladling}.
图无处不在,gnn是一个强大的神经网络家族,用于在图上学习。尽管它们很受欢迎,但通过加深或扩大来缩放gnn存在普遍存在的不健康梯度、过度平滑、信息压缩等问题,这些问题往往导致性能低于标准。在这项工作中,我们感兴趣的是探索一种原则性的方法来扩展gnn的容量,而不需要加深或扩大,这可以提高其在多个小图和大图上的性能。最近模型汤的有趣现象表明,多个大语言预训练模型的微调权重可以合并到一个更好的最小值,我们认为可以利用模型汤的基本原理来缓解上述gnn缩放过程中的内存瓶颈和可训练性问题。更具体地说,我们建议不深化或扩大当前的gnn,而是提出一个以数据为中心的视角,为gnn量身定制模型汤,即构建强大的gnn。通过分割庞大的图数据,我们在没有任何中间通信的情况下构建多个独立且并行训练的较弱gnn(汤成分),并使用贪婪插值汤程序组合它们的强度以达到最先进的性能。与Jiong et al. 2023等并行分布式GNN训练作品相比,我们通过每个epoch采样不同的子图来训练每个汤成分,并且它们各自的子模型只有在完全训练后才合并(而不是中间)。此外,我们通过利用可以处理大型图的最先进的图采样和图划分方法,提供了各种各样的模型汤制备技术。代码可在url{https://github.com/VITA-Group/graph_ladling}获得。
{"title":"Graph Ladling: Shockingly Simple Parallel GNN Training without Intermediate Communication","authors":"A. Jaiswal, Shiwei Liu, Tianlong Chen, Ying Ding, Zhangyang Wang","doi":"10.48550/arXiv.2306.10466","DOIUrl":"https://doi.org/10.48550/arXiv.2306.10466","url":null,"abstract":"Graphs are omnipresent and GNNs are a powerful family of neural networks for learning over graphs. Despite their popularity, scaling GNNs either by deepening or widening suffers from prevalent issues of unhealthy gradients, over-smoothening, information squashing, which often lead to sub-standard performance. In this work, we are interested in exploring a principled way to scale GNNs capacity without deepening or widening, which can improve its performance across multiple small and large graphs. Motivated by the recent intriguing phenomenon of model soups, which suggest that fine-tuned weights of multiple large-language pre-trained models can be merged to a better minima, we argue to exploit the fundamentals of model soups to mitigate the aforementioned issues of memory bottleneck and trainability during GNNs scaling. More specifically, we propose not to deepen or widen current GNNs, but instead present a data-centric perspective of model soups tailored for GNNs, i.e., to build powerful GNNs. By dividing giant graph data, we build multiple independently and parallelly trained weaker GNNs (soup ingredient) without any intermediate communication, and combine their strength using a greedy interpolation soup procedure to achieve state-of-the-art performance. Compared to concurrent distributed GNN training works such as Jiong et. al. 2023, we train each soup ingredient by sampling different subgraphs per epoch and their respective sub-models are merged only after being fully trained (rather than intermediately so). Moreover, we provide a wide variety of model soup preparation techniques by leveraging state-of-the-art graph sampling and graph partitioning approaches that can handle large graphs. Codes are available at: url{https://github.com/VITA-Group/graph_ladling}.","PeriodicalId":74529,"journal":{"name":"Proceedings of the ... International Conference on Machine Learning. International Conference on Machine Learning","volume":"71 1","pages":"14679-14690"},"PeriodicalIF":0.0,"publicationDate":"2023-06-18","publicationTypes":"Journal Article","fieldsOfStudy":null,"isOpenAccess":false,"openAccessPdf":"","citationCount":null,"resultStr":null,"platform":"Semanticscholar","paperid":"83885598","PeriodicalName":null,"FirstCategoryId":null,"ListUrlMain":null,"RegionNum":0,"RegionCategory":"","ArticlePicture":[],"TitleCN":null,"AbstractTextCN":null,"PMCID":"","EPubDate":null,"PubModel":null,"JCR":null,"JCRName":null,"Score":null,"Total":0}
引用次数: 1
Weakly Supervised Regression with Interval Targets 具有区间目标的弱监督回归
Xi Cheng, Yuzhou Cao, Ximing Li, Bo An, Lei Feng
This paper investigates an interesting weakly supervised regression setting called regression with interval targets (RIT). Although some of the previous methods on relevant regression settings can be adapted to RIT, they are not statistically consistent, and thus their empirical performance is not guaranteed. In this paper, we provide a thorough study on RIT. First, we proposed a novel statistical model to describe the data generation process for RIT and demonstrate its validity. Second, we analyze a simple selection method for RIT, which selects a particular value in the interval as the target value to train the model. Third, we propose a statistically consistent limiting method for RIT to train the model by limiting the predictions to the interval. We further derive an estimation error bound for our limiting method. Finally, extensive experiments on various datasets demonstrate the effectiveness of our proposed method.
本文研究了一种有趣的弱监督回归设置,称为区间目标回归。虽然之前的一些方法在相关回归设置上可以适用于RIT,但它们在统计上并不一致,因此不能保证它们的经验性能。在本文中,我们对RIT进行了深入的研究。首先,我们提出了一个新的统计模型来描述RIT的数据生成过程,并证明了它的有效性。其次,我们分析了一种简单的RIT选择方法,即在区间中选择一个特定的值作为训练模型的目标值。第三,我们提出了一种统计一致的RIT限制方法,通过将预测限制在区间内来训练模型。进一步导出了该方法的估计误差界。最后,在各种数据集上的大量实验证明了我们提出的方法的有效性。
{"title":"Weakly Supervised Regression with Interval Targets","authors":"Xi Cheng, Yuzhou Cao, Ximing Li, Bo An, Lei Feng","doi":"10.48550/arXiv.2306.10458","DOIUrl":"https://doi.org/10.48550/arXiv.2306.10458","url":null,"abstract":"This paper investigates an interesting weakly supervised regression setting called regression with interval targets (RIT). Although some of the previous methods on relevant regression settings can be adapted to RIT, they are not statistically consistent, and thus their empirical performance is not guaranteed. In this paper, we provide a thorough study on RIT. First, we proposed a novel statistical model to describe the data generation process for RIT and demonstrate its validity. Second, we analyze a simple selection method for RIT, which selects a particular value in the interval as the target value to train the model. Third, we propose a statistically consistent limiting method for RIT to train the model by limiting the predictions to the interval. We further derive an estimation error bound for our limiting method. Finally, extensive experiments on various datasets demonstrate the effectiveness of our proposed method.","PeriodicalId":74529,"journal":{"name":"Proceedings of the ... International Conference on Machine Learning. International Conference on Machine Learning","volume":"30 1","pages":"5428-5448"},"PeriodicalIF":0.0,"publicationDate":"2023-06-18","publicationTypes":"Journal Article","fieldsOfStudy":null,"isOpenAccess":false,"openAccessPdf":"","citationCount":null,"resultStr":null,"platform":"Semanticscholar","paperid":"89140738","PeriodicalName":null,"FirstCategoryId":null,"ListUrlMain":null,"RegionNum":0,"RegionCategory":"","ArticlePicture":[],"TitleCN":null,"AbstractTextCN":null,"PMCID":"","EPubDate":null,"PubModel":null,"JCR":null,"JCRName":null,"Score":null,"Total":0}
引用次数: 1
Instant Soup: Cheap Pruning Ensembles in A Single Pass Can Draw Lottery Tickets from Large Models 速溶汤:一次通过廉价的修剪组合可以从大型模型中抽取彩票
A. Jaiswal, Shiwei Liu, Tianlong Chen, Ying Ding, Zhangyang Wang
Large pre-trained transformers have been receiving explosive attention in the past few years, due to their wide adaptability for numerous downstream applications via fine-tuning, but their exponentially increasing parameter counts are becoming a primary hurdle to even just fine-tune them without industry-standard hardware. Recently, Lottery Ticket Hypothesis (LTH) and its variants, have been exploited to prune these large pre-trained models generating subnetworks that can achieve similar performance as their dense counterparts, but LTH pragmatism is enormously inhibited by repetitive full training and pruning routine of iterative magnitude pruning (IMP) which worsens with increasing model size. Motivated by the recent observations of model soups, which suggest that fine-tuned weights of multiple models can be merged to a better minima, we propose Instant Soup Pruning (ISP) to generate lottery ticket quality subnetworks, using a fraction of the original IMP cost by replacing the expensive intermediate pruning stages of IMP with computationally efficient weak mask generation and aggregation routine. More specifically, during the mask generation stage, ISP takes a small handful of iterations using varying training protocols and data subsets to generate many weak and noisy subnetworks, and superpose them to average out the noise creating a high-quality denoised subnetwork. Our extensive experiments and ablation on two popular large-scale pre-trained models: CLIP (unexplored in pruning till date) and BERT across multiple benchmark vision and language datasets validate the effectiveness of ISP compared to several state-of-the-art pruning methods. Codes are available at: url{https://github.com/VITA-Group/instant_soup}
大型预训练变压器在过去几年中受到了爆炸性的关注,因为它们通过微调对许多下游应用具有广泛的适应性,但是它们指数级增长的参数数量正在成为一个主要障碍,即使没有工业标准硬件也要对它们进行微调。最近,彩票假设(LTH)及其变体被用于修剪这些大型预训练模型,生成的子网络可以达到与其密集对应的子网络相似的性能,但LTH的实用主义受到重复的完整训练和迭代量级修剪(IMP)的修剪程序的极大抑制,该程序随着模型大小的增加而恶化。最近对模型汤的观察表明,多个模型的微调权重可以合并到一个更好的最小值,我们提出了即时汤修剪(ISP)来生成彩票质量的子网,使用原始IMP成本的一小部分,通过使用计算效率高的弱掩模生成和聚合程序取代IMP昂贵的中间修剪阶段。更具体地说,在掩码生成阶段,ISP使用不同的训练协议和数据子集进行少量迭代,以生成许多弱和有噪声的子网,并将它们叠加以平均噪声,从而创建高质量的去噪子网。我们对两种流行的大规模预训练模型进行了广泛的实验和研究:CLIP(迄今为止尚未在修剪方面进行过探索)和BERT,跨多个基准视觉和语言数据集验证了ISP与几种最先进的修剪方法相比的有效性。守则可于以下网址取得: url{https://github.com/VITA-Group/instant_soup}
{"title":"Instant Soup: Cheap Pruning Ensembles in A Single Pass Can Draw Lottery Tickets from Large Models","authors":"A. Jaiswal, Shiwei Liu, Tianlong Chen, Ying Ding, Zhangyang Wang","doi":"10.48550/arXiv.2306.10460","DOIUrl":"https://doi.org/10.48550/arXiv.2306.10460","url":null,"abstract":"Large pre-trained transformers have been receiving explosive attention in the past few years, due to their wide adaptability for numerous downstream applications via fine-tuning, but their exponentially increasing parameter counts are becoming a primary hurdle to even just fine-tune them without industry-standard hardware. Recently, Lottery Ticket Hypothesis (LTH) and its variants, have been exploited to prune these large pre-trained models generating subnetworks that can achieve similar performance as their dense counterparts, but LTH pragmatism is enormously inhibited by repetitive full training and pruning routine of iterative magnitude pruning (IMP) which worsens with increasing model size. Motivated by the recent observations of model soups, which suggest that fine-tuned weights of multiple models can be merged to a better minima, we propose Instant Soup Pruning (ISP) to generate lottery ticket quality subnetworks, using a fraction of the original IMP cost by replacing the expensive intermediate pruning stages of IMP with computationally efficient weak mask generation and aggregation routine. More specifically, during the mask generation stage, ISP takes a small handful of iterations using varying training protocols and data subsets to generate many weak and noisy subnetworks, and superpose them to average out the noise creating a high-quality denoised subnetwork. Our extensive experiments and ablation on two popular large-scale pre-trained models: CLIP (unexplored in pruning till date) and BERT across multiple benchmark vision and language datasets validate the effectiveness of ISP compared to several state-of-the-art pruning methods. Codes are available at: url{https://github.com/VITA-Group/instant_soup}","PeriodicalId":74529,"journal":{"name":"Proceedings of the ... International Conference on Machine Learning. International Conference on Machine Learning","volume":"10 1","pages":"14691-14701"},"PeriodicalIF":0.0,"publicationDate":"2023-06-18","publicationTypes":"Journal Article","fieldsOfStudy":null,"isOpenAccess":false,"openAccessPdf":"","citationCount":null,"resultStr":null,"platform":"Semanticscholar","paperid":"84829455","PeriodicalName":null,"FirstCategoryId":null,"ListUrlMain":null,"RegionNum":0,"RegionCategory":"","ArticlePicture":[],"TitleCN":null,"AbstractTextCN":null,"PMCID":"","EPubDate":null,"PubModel":null,"JCR":null,"JCRName":null,"Score":null,"Total":0}
引用次数: 3
期刊
Proceedings of the ... International Conference on Machine Learning. International Conference on Machine Learning
全部 Acc. Chem. Res. ACS Applied Bio Materials ACS Appl. Electron. Mater. ACS Appl. Energy Mater. ACS Appl. Mater. Interfaces ACS Appl. Nano Mater. ACS Appl. Polym. Mater. ACS BIOMATER-SCI ENG ACS Catal. ACS Cent. Sci. ACS Chem. Biol. ACS Chemical Health & Safety ACS Chem. Neurosci. ACS Comb. Sci. ACS Earth Space Chem. ACS Energy Lett. ACS Infect. Dis. ACS Macro Lett. ACS Mater. Lett. ACS Med. Chem. Lett. ACS Nano ACS Omega ACS Photonics ACS Sens. ACS Sustainable Chem. Eng. ACS Synth. Biol. Anal. Chem. BIOCHEMISTRY-US Bioconjugate Chem. BIOMACROMOLECULES Chem. Res. Toxicol. Chem. Rev. Chem. Mater. CRYST GROWTH DES ENERG FUEL Environ. Sci. Technol. Environ. Sci. Technol. Lett. Eur. J. Inorg. Chem. IND ENG CHEM RES Inorg. Chem. J. Agric. Food. Chem. J. Chem. Eng. Data J. Chem. Educ. J. Chem. Inf. Model. J. Chem. Theory Comput. J. Med. Chem. J. Nat. Prod. J PROTEOME RES J. Am. Chem. Soc. LANGMUIR MACROMOLECULES Mol. Pharmaceutics Nano Lett. Org. Lett. ORG PROCESS RES DEV ORGANOMETALLICS J. Org. Chem. J. Phys. Chem. J. Phys. Chem. A J. Phys. Chem. B J. Phys. Chem. C J. Phys. Chem. Lett. Analyst Anal. Methods Biomater. Sci. Catal. Sci. Technol. Chem. Commun. Chem. Soc. Rev. CHEM EDUC RES PRACT CRYSTENGCOMM Dalton Trans. Energy Environ. Sci. ENVIRON SCI-NANO ENVIRON SCI-PROC IMP ENVIRON SCI-WAT RES Faraday Discuss. Food Funct. Green Chem. Inorg. Chem. Front. Integr. Biol. J. Anal. At. Spectrom. J. Mater. Chem. A J. Mater. Chem. B J. Mater. Chem. C Lab Chip Mater. Chem. Front. Mater. Horiz. MEDCHEMCOMM Metallomics Mol. Biosyst. Mol. Syst. Des. Eng. Nanoscale Nanoscale Horiz. Nat. Prod. Rep. New J. Chem. Org. Biomol. Chem. Org. Chem. Front. PHOTOCH PHOTOBIO SCI PCCP Polym. Chem.
×
引用
GB/T 7714-2015
复制
MLA
复制
APA
复制
导出至
BibTeX EndNote RefMan NoteFirst NoteExpress
×
0
微信
客服QQ
Book学术公众号 扫码关注我们
反馈
×
意见反馈
请填写您的意见或建议
请填写您的手机或邮箱
×
提示
您的信息不完整,为了账户安全,请先补充。
现在去补充
×
提示
您因"违规操作"
具体请查看互助需知
我知道了
×
提示
现在去查看 取消
×
提示
确定
Book学术官方微信
Book学术文献互助
Book学术文献互助群
群 号:481959085
Book学术
文献互助 智能选刊 最新文献 互助须知 联系我们:info@booksci.cn
Book学术提供免费学术资源搜索服务,方便国内外学者检索中英文文献。致力于提供最便捷和优质的服务体验。
Copyright © 2023 Book学术 All rights reserved.
ghs 京公网安备 11010802042870号 京ICP备2023020795号-1