Skip to content

Graph wavelets API documentation

diffusion_wavelets 

diffusion_wavelets(T, n_scales)

Compute diffusion wavelet filter bank

Computes diffusion wavelets from from input diffusion matrix. Implementation based off the algorithm originally described in Coifman et. al 2006.

Parameters:

Name Type Description Default
T torch.Tensor

Input diffusion matrix computed from adjacency matrix

 required
n_scales int

Number of scales to use in wavelet transform

 required

Returns:

Name Type Description
phi torch.Tensor

wavelet filter bank
Source code in gsxform/wavelets.py 
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
def diffusion_wavelets(T: torch.Tensor, n_scales: int) -> torch.Tensor:
    """Compute diffusion wavelet filter bank

    Computes diffusion wavelets from from input diffusion matrix.
    Implementation based off the algorithm originally described in
    Coifman et. al 2006.

    Parameters
    ----------
    T: torch.Tensor
        Input diffusion matrix computed from adjacency matrix
    n_scales: int
        Number of scales to use in wavelet transform

    Returns
    -------
    phi: torch.Tensor
        wavelet filter bank
    """

    # make n_node x n_node identity matrix
    I_N = torch.eye(T.shape[1])

    # compute zero-eth order (J=0) wavelet filter
    # one half the normalized laplacian operator 1/2(I-D^-1/2WD^-1/2)
    psi = I_N - T

    for jj in range(1, n_scales):
        # compute jth diffusion operator (wavelet kernel)
        T_j = torch.matrix_power(T, 2 ** (jj - 1))
        # compute jth wavelet filter via matmul
        # psi_j = torch.einsum("b n m, b n m -> b n m", T_j, (I_N - T_j))
        psi_j = torch.matmul(T_j, (I_N - T_j))
        # append wavelets
        psi = torch.cat((psi, psi_j), axis=0)

    psi = rearrange(psi, "(b ns) ni nj -> b ns ni nj", ns=n_scales)

    return psi

tighthann_wavelets 

tighthann_wavelets(W_adj, n_scales, kernel)

Computes spectrum adapted tight Hann wavelets. Based of algorithm described in Shuman et. al 2015.

Parameters:

Name Type Description Default
W_adj torch.Tensor

Input batch of adjacency matricies

 required
n_scales int

Number of scales to use in wavelet transform

 required
kernel TightHannKernel

Adaptive kernel used in wavelet transform.

 required

Returns:

Name Type Description
psi torch.Tensor

wavelet filter bank
Source code in gsxform/wavelets.py 
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
def tighthann_wavelets(
    W_adj: torch.Tensor, n_scales: int, kernel: TightHannKernel
) -> torch.Tensor:
    """Computes spectrum adapted tight Hann wavelets. Based
    of algorithm described in Shuman et. al 2015.

    Parameters
    ----------
    W_adj: torch.Tensor
        Input batch of adjacency matricies
    n_scales: int
        Number of scales to use in wavelet transform
    kernel: TightHannKernel
        Adaptive kernel used in wavelet transform.

    Returns
    -------
    psi: torch.Tensor
        wavelet filter bank

    """
    E, V = compute_spectra(W_adj)

    V_herm = rearrange(V, "b ni nj -> b nj ni")  # hermetian transpose

    # compute wavelet coeffs
    psi = torch.empty(V.shape[0], 0, V.shape[1], V.shape[2])
    for jj in range(0, n_scales):

        # compute adapted kernel
        adapted_kernel = kernel.get_adapted_kernel(E, jj + 1)
        phi = torch.diag_embed(adapted_kernel)

        # compute jth wavelet filter via matmul
        psi_j = V.matmul(phi).matmul(V_herm)

        # append wavelets
        psi_j = rearrange(psi_j, "b n m -> b 1 n m")

        psi = torch.cat((psi, psi_j), axis=1)

    return psi