add and export function dmatrix_ldb_flatten()

Author: haotian127

src/MultiscaleGraphSignalTransforms.jl

@@ -19,7 +19,7 @@ using Reexport
 @reexport using .GraphSignal, .GraphPartition, .BasisSpecification, .GHWT, .GHWT_2d, .GHWT_tf_1d, .GHWT_tf_2d, .HGLET
 
 export dvec2dmatrix, dmatrix2dvec, levlist2levlengths!, bsfull, bs_haar, bs_level, bs_walsh, dvec_Threshold, rs_to_region, GraphSig_Plot, gplot, gplot!, partition_fiedler
-export cost_functional, dmatrix_flatten
+export cost_functional, dmatrix_flatten, dmatrix_ldb_flatten
 
 ## export functions of NGWP.jl
 using LinearAlgebra, SparseArrays, LightGraphs, SimpleWeightedGraphs, Clustering

src/utils.jl

@@ -167,6 +167,71 @@ function dmatrix_flatten(dmatrix::Array{Float64,3}, flatten::Any)
     return dmatrix
 end # of function dmatrix_flatten!
 
+"""
+function ldb_discriminant_measure(p::Vector{Float64}, q::Vector{Float64}; dm::Symbol = :KLdivergence)
+
+Discriminat measure in LDB.
+
+### Input Arguments
+* `p,q::Vector{Float64}`: probability mass functions.
+* `dm::Symbol`: discriminant measure. Options: `:KLdivergence`(default),
+    `:Jdivergence`, `:l1`, `:l2`, and `:Hellinger`.
+"""
+function ldb_discriminant_measure(p::Vector{Float64}, q::Vector{Float64}; dm::Symbol = :KLdivergence)
+    @assert all(p .>= 0) && all(q .>= 0)
+    @assert length(p) == length(q)
+    if dm == :KLdivergence
+        ind = findall((p .> 0) .& (q .> 0))
+        return Distances.kl_divergence(p[ind], q[ind])
+    elseif dm == :Jdivergence
+        ind = findall((p .> 0) .& (q .> 0))
+        return Distances.kl_divergence(p[ind], q[ind]) + Distances.kl_divergence(q[ind], p[ind])
+    elseif dm == :l1
+        return norm(p - q, 1)
+    elseif dm == :l2
+        return norm(p - q, 2)
+    elseif dm == :Hellinger
+        return hellinger(p, q)
+    else
+        error("This discriminat measure $(dm) is not supported! ")
+    end
+end
+
+"""
+    function dmatrix_ldb_flatten(dmatrix::Array{Float64,3}...; dm::Symbol = :KLdivergence)
+
+Flatten dmatrix using the LDB method; after this function is called, it becomes
+the size of (~, ~, 1). Example usage: dmatrix_ldb_flatten(dmatrix1, dmatrix2, dmatrix3),
+each argument is the expansion coefficient matrix of a class of signals.
+
+### Input Arguments
+* `dmatrix::Array{Float64,3}`: the matrix of LDB expansion coefficients in one class
+* `dm::Symbol`: discriminant measure. Options: `:KLdivergence`(default),
+    `:Jdivergence`, `:l1`, `:l2`, and `:Hellinger`.
+"""
+function dmatrix_ldb_flatten(dmatrix::Array{Float64,3}...; dm::Symbol = :KLdivergence)
+    C = length(dmatrix) # number of signal classes
+    if C < 2
+        error("Input should contain at least two classes of signals.")
+    end
+    N, jmax, _ = Base.size(dmatrix[1])
+    res = zeros(N, jmax)
+    for u = 1:(C - 1), v = (i + 1):C
+        for j = 1:jmax
+            for i = 1:N
+                h1 = ash(dmatrix[u][i, j, :])
+                p = h1.density / norm(h1.density, 1)
+                h2 = ash(dmatrix[v][i, j, :])
+                q = h2.density / norm(h2.density, 1)
+                res[i, j] += ldb_discriminant_measure(p, q; dm = dm)
+            end
+        end
+    end
+    res = reshape(res[:, :, 1], N, jmax, 1)
+    return res
+end
+
+
 """
     costfun = cost_functional(cfspec::Any)