-
Notifications
You must be signed in to change notification settings - Fork 309
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge pull request #169 from Yelrose/pgl-2.0.0a
Pgl 2.0.0a
- Loading branch information
Showing
390 changed files
with
775,597 additions
and
1,458 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,8 @@ | ||
pgl.dataset: Some benchmark datasets. | ||
================================================= | ||
|
||
.. automodule:: pgl.dataset | ||
:members: | ||
:undoc-members: | ||
:show-inheritance: | ||
|
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,8 @@ | ||
pgl.graph: Graph Storage | ||
=============================== | ||
|
||
.. automodule:: pgl.graph | ||
:members: | ||
:undoc-members: | ||
:show-inheritance: | ||
|
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,12 @@ | ||
pgl.message | ||
============== | ||
|
||
|
||
The Message Implement for recv function | ||
-------------------------------------- | ||
|
||
.. automodule:: pgl.message | ||
:members: | ||
:undoc-members: | ||
:show-inheritance: | ||
|
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,13 @@ | ||
pgl.nn.functional | ||
========================= | ||
|
||
|
||
|
||
Graph Level Function | ||
---------------------------------- | ||
|
||
.. automodule:: pgl.nn.functional.graph_op | ||
:members: | ||
:undoc-members: | ||
:show-inheritance: | ||
|
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,19 @@ | ||
pgl.nn: Predefined graph neural networks layers. | ||
============== | ||
|
||
|
||
Graph Convolution Layers | ||
------------------------ | ||
|
||
.. automodule:: pgl.nn.conv | ||
:members: | ||
:undoc-members: | ||
:show-inheritance: | ||
|
||
Graph Pooling Layers | ||
------------------------ | ||
|
||
.. automodule:: pgl.nn.pool | ||
:members: | ||
:undoc-members: | ||
:show-inheritance: |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -1,13 +1,12 @@ | ||
API Reference | ||
============= | ||
=========== | ||
|
||
.. toctree:: | ||
|
||
pgl.graph | ||
pgl.graph_wrapper | ||
pgl.layers | ||
pgl.data_loader | ||
pgl.utils.paddle_helper | ||
pgl.utils.mp_reader | ||
pgl.heter_graph | ||
pgl.heter_graph_wrapper | ||
graph | ||
sampling | ||
nn | ||
nn.functional | ||
dataset | ||
utils.mp_reader | ||
message |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,13 @@ | ||
pgl.sampling | ||
========================= | ||
|
||
|
||
|
||
Graph Sampling Function | ||
---------------------------------- | ||
|
||
.. automodule:: pgl.sampling | ||
:members: | ||
:undoc-members: | ||
:show-inheritance: | ||
|
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1 @@ | ||
.. mdinclude:: ../../../examples/citation_benchmark/README.md |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1 @@ | ||
.. mdinclude:: ../../../examples/gin/README.md |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,145 @@ | ||
# Paddle Graph Learning (PGL) | ||
|
||
Paddle Graph Learning (PGL) is an efficient and flexible graph learning framework based on [PaddlePaddle](https://github.com/PaddlePaddle/Paddle). | ||
|
||
|
||
<div /> | ||
<div align=center><img src="../_static/framework_of_pgl.png" width="700"></div> | ||
<center>The Framework of Paddle Graph Learning (PGL)</center> | ||
<div /> | ||
|
||
The newly released PGL supports heterogeneous graph learning on both walk based paradigm and message-passing based paradigm by providing MetaPath sampling and Message Passing mechanism on heterogeneous graph. Furthermor, The newly released PGL also support distributed graph storage and some distributed training algorithms, such as distributed deep walk and distributed graphsage. Combined with the PaddlePaddle deep learning framework, we are able to support both graph representation learning models and graph neural networks, and thus our framework has a wide range of graph-based applications. | ||
|
||
## Highlight: Efficiency - Support Scatter-Gather and LodTensor Message Passing | ||
|
||
One of the most important benefits of graph neural networks compared to other models is the ability to use node-to-node connectivity information, but coding the communication between nodes is very cumbersome. At PGL we adopt **Message Passing Paradigm** similar to DGL to help to build a customize graph neural network easily. Users only need to write ``send`` and ``recv`` functions to easily implement a simple GCN. As shown in the following figure, for the first step the send function is defined on the edges of the graph, and the user can customize the send function $\phi^e$ to send the message from the source to the target node. For the second step, the recv function $\phi^v$ is responsible for aggregating $\oplus$ messages together from different sources. | ||
|
||
<div /> | ||
<div align=center><img src="../_static/message_passing_paradigm.png" width="700"></div> | ||
<center>The basic idea of message passing paradigm</center> | ||
<div /> | ||
|
||
As shown in the left of the following figure, to adapt general user-defined message aggregate functions, DGL uses the degree bucketing method to combine nodes with the same degree into a batch and then apply an aggregate function $\oplus$ on each batch serially. For our PGL UDF aggregate function, we organize the message as a [LodTensor](http://www.paddlepaddle.org/documentation/docs/en/1.4/user_guides/howto/basic_concept/lod_tensor_en.html) in [PaddlePaddle](https://github.com/PaddlePaddle/Paddle) taking the message as variable length sequences. And we **utilize the features of LodTensor in Paddle to obtain fast parallel aggregation**. | ||
|
||
|
||
<div/> | ||
<div align=center><img src="../_static/parallel_degree_bucketing.png" width="750"></div> | ||
<center>The parallel degree bucketing of PGL</center> | ||
<div/> | ||
|
||
|
||
Users only need to call the ``sequence_ops`` functions provided by Paddle to easily implement efficient message aggregation. For examples, using ``sequence_pool`` to sum the neighbor message. | ||
```python | ||
import paddle.fluid as fluid | ||
def recv(msg): | ||
return fluid.layers.sequence_pool(msg, "sum") | ||
``` | ||
|
||
|
||
Although DGL does some kernel fusion optimization for general sum, max and other aggregate functions with scatter-gather. For **complex user-defined functions** with degree bucketing algorithm, the serial execution for each degree bucket cannot take full advantage of the performance improvement provided by GPU. However, operations on the PGL LodTensor-based message is performed in parallel, which can fully utilize GPU parallel optimization. In our experiments, PGL can reach up to 13 times the speed of DGL with complex user-defined functions. Even without scatter-gather optimization, PGL still has excellent performance. Of course, we still provide build-in scatter-optimized message aggregation functions. | ||
|
||
### Performance | ||
|
||
|
||
We test all the following GNN algorithms with Tesla V100-SXM2-16G running for 200 epochs to get average speeds. And we report the accuracy on test dataset without early stoppping. | ||
|
||
| Dataset | Model | PGL Accuracy | PGL speed (epoch time) | DGL 0.3.0 speed (epoch time) | | ||
| -------- | ----- | ----------------- | ------------ | ------------------------------------ | | ||
| Cora | GCN |81.75% | 0.0047s | **0.0045s** | | ||
| Cora | GAT | 83.5% | **0.0119s** | 0.0141s | | ||
| Pubmed | GCN |79.2% |**0.0049s** |0.0051s | | ||
| Pubmed | GAT | 77% |0.0193s|**0.0144s**| | ||
| Citeseer | GCN |70.2%| **0.0045** |0.0046s| | ||
| Citeseer | GAT |68.8%| **0.0124s** |0.0139s| | ||
|
||
|
||
If we use complex user-defined aggregation like [GraphSAGE-LSTM](https://cs.stanford.edu/people/jure/pubs/graphsage-nips17.pdf) that aggregates neighbor features with LSTM ignoring the order of recieved messages, the optimized message-passing in DGL will be forced to degenerate into degree bucketing scheme. The speed performance will be much slower than the one implemented in PGL. Performances may be various with different scale of the graph, in our experiments, PGL can reach up to 13 times the speed of DGL. | ||
|
||
| Dataset | PGL speed (epoch time) | DGL 0.3.0 speed (epoch time) | Speed up| | ||
| -------- | ------------ | ------------------------------------ |----| | ||
| Cora | **0.0186s** | 0.1638s | 8.80x| | ||
| Pubmed | **0.0388s** |0.5275s | 13.59x| | ||
| Citeseer | **0.0150s** | 0.1278s | 8.52x | | ||
|
||
## Highlight: Flexibility - Natively Support Heterogeneous Graph Learning | ||
|
||
Graph can conveniently represent the relation between things in the real world, but the categories of things and the relation between things are various. Therefore, in the heterogeneous graph, we need to distinguish the node types and edge types in the graph network. PGL models heterogeneous graphs that contain multiple node types and multiple edge types, and can describe complex connections between different types. | ||
|
||
### Support meta path walk sampling on heterogeneous graph | ||
|
||
<div/> | ||
<div align=center><img src="../_static/metapath_sampling.png" width="750"></div> | ||
<center>The metapath sampling in heterogeneous graph</center> | ||
<div/> | ||
The left side of the figure above describes a shopping social network. The nodes above have two categories of users and goods, and the relations between users and users, users and goods, and goods and goods. The right of the above figure is a simple sampling process of MetaPath. When you input any MetaPath as UPU (user-product-user), you will find the following results | ||
<div/> | ||
<div align=center><img src="../_static/metapath_result.png" width="300"></div> | ||
<center>The metapath result</center> | ||
<div/> | ||
Then on this basis, and introducing word2vec and other methods to support learning metapath2vec and other algorithms of heterogeneous graph representation. | ||
|
||
### Support Message Passing mechanism on heterogeneous graph | ||
|
||
<div/> | ||
<div align=center><img src="../_static/him_message_passing.png" width="750"></div> | ||
<center>The message passing mechanism on heterogeneous graph</center> | ||
<div/> | ||
Because of the different node types on the heterogeneous graph, the message delivery is also different. As shown on the left, it has five neighbors, belonging to two different node types. As shown on the right of the figure above, nodes belonging to different types need to be aggregated separately during message delivery, and then merged into the final message to update the target node. On this basis, PGL supports heterogeneous graph algorithms based on message passing, such as GATNE and other algorithms. | ||
|
||
|
||
## Large-Scale: Support distributed graph storage and distributed training algorithms | ||
In most cases of large-scale graph learning, we need distributed graph storage and distributed training support. As shown in the following figure, PGL provided a general solution of large-scale training, we adopted [PaddleFleet](https://github.com/PaddlePaddle/Fleet) as our distributed parameter servers, which supports large scale distributed embeddings and a lightweighted distributed storage engine so tcan easily set up a large scale distributed training algorithm with MPI clusters. | ||
|
||
<div/> | ||
<div align=center><img src="../_static/distributed_frame.png" width="750"></div> | ||
<center>The distributed frame of PGL</center> | ||
<div/> | ||
|
||
|
||
## Model Zoo | ||
The following are 13 graph learning models that have been implemented in the framework. | ||
|
||
|Model | feature | | ||
|---|---| | ||
| [GCN](examples/gcn_examples.html)| Graph Convolutional Neural Networks | | ||
| [GAT](examples/gat_examples.html)| Graph Attention Network | | ||
| [GraphSage](examples/graphsage_examples.html)|Large-scale graph convolution network based on neighborhood sampling| | ||
| [unSup-GraphSage](examples/unsup_graphsage_examples.html) | Unsupervised GraphSAGE | | ||
| [LINE](examples/line_examples.html)| Representation learning based on first-order and second-order neighbors | | ||
| [DeepWalk](examples/distribute_deepwalk_examples.html)| Representation learning by DFS random walk | | ||
| [MetaPath2Vec](examples/metapath2vec_examples.html)| Representation learning based on metapath | | ||
| [Node2Vec](examples/node2vec_examples.html)| The representation learning Combined with DFS and BFS | | ||
| [Struct2Vec](examples/strucvec_examples.html)| Representation learning based on structural similarity | | ||
| [SGC](examples/sgc_examples.html)| Simplified graph convolution neural network | | ||
| [GES](examples/ges_examples.html)| The graph represents learning method with node features | | ||
| [DGI](examples/dgi_examples.html)| Unsupervised representation learning based on graph convolution network | | ||
| [GATNE](examples/gatne_examples.html)| Representation Learning of Heterogeneous Graph based on MessagePassing | | ||
|
||
The above models consists of three parts, namely, graph representation learning, graph neural network and heterogeneous graph learning, which are also divided into graph representation learning and graph neural network. | ||
|
||
## System requirements | ||
|
||
PGL requires: | ||
|
||
* paddle >= 1.6 | ||
* cython | ||
|
||
|
||
PGL supports both Python 2 & 3 | ||
|
||
|
||
## Installation | ||
|
||
You can simply install it via pip. | ||
|
||
```sh | ||
pip install pgl | ||
``` | ||
|
||
## The Team | ||
|
||
PGL is developed and maintained by NLP and Paddle Teams at Baidu | ||
|
||
## License | ||
|
||
PGL uses Apache License 2.0. |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1 @@ | ||
.. mdinclude:: introduction.md |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.