File size: 5,638 Bytes
5c61046 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 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 54 55 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 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 | # tda_braid_map.jl β Barcodes β BraidWords on Heavy-Hex
module TDABraidMap
using LinearAlgebra
using Random
export barcode_to_braid_word, feature_diff_to_braid, heavy_hex_braid_generators
export pairwise_braid_words
# βββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββ
# Types (imported from YaoTypes in full build)
# βββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββ
struct BraidWord
generators::Vector{Int}
edge_indices::Vector{Int}
n_strands::Int
function BraidWord(gens::Vector{Int}, edges::Vector{Int}, n_strands::Int)
@assert length(gens) == length(edges)
new(gens, edges, n_strands)
end
end
BraidWord(n_strands::Int) = BraidWord(Int[], Int[], n_strands)
struct PersistenceInterval
dim::Int
birth::Float64
death::Float64
end
struct Barcode
H0::Vector{PersistenceInterval}
H1::Vector{PersistenceInterval}
end
const HERON_EDGES_0 = [
(0, 1), (1, 2),
(0, 3), (1, 3), (1, 4), (2, 4), (2, 5),
(3, 4), (4, 5), (5, 6),
(3, 7), (4, 7), (4, 8), (5, 8), (5, 9), (6, 9),
(7, 8), (8, 9)
]
const HERON_EDGE_INDEX = Dict(edge => i for (i, edge) in enumerate(HERON_EDGES_0))
# βββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββ
# Heavy-Hex Braid Generators
# βββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββ
function heavy_hex_braid_generators(n_strands::Int)::Dict{Int, Tuple{Int,Int}}
gens = Dict{Int, Tuple{Int,Int}}()
for i in 1:min(n_strands-1, length(HERON_EDGES_0))
gens[i] = HERON_EDGES_0[i]
end
return gens
end
# βββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββ
# Barcode β Braid Word
# βββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββ
"""
barcode_to_braid_word(bc::Barcode, n_strands::Int; persistence_threshold=0.1)
Map persistent homology intervals to Artin generators.
High-persistence H1 features β over-crossings (Ο)
Low-persistence / noise β under-crossings (Οβ»ΒΉ) or identity
"""
function barcode_to_braid_word(bc::Barcode, n_strands::Int;
persistence_threshold::Float64=0.1)::BraidWord
generators = Int[]
edge_indices = Int[]
gens_map = heavy_hex_braid_generators(n_strands)
n_gens = length(gens_map)
for (idx, intv) in enumerate(bc.H1)
pers = intv.death - intv.birth
if pers < persistence_threshold
continue
end
gen_idx = (idx - 1) % n_gens + 1
edge = gens_map[gen_idx]
edge_idx = HERON_EDGE_INDEX[edge]
sign = (idx % 2 == 1) ? 1 : -1
push!(generators, sign * gen_idx)
push!(edge_indices, edge_idx)
end
if isempty(generators)
return BraidWord(n_strands)
end
BraidWord(generators, edge_indices, n_strands)
end
"""
feature_diff_to_braid(x, xβ², n_strands; epsilon=0.5)
Direct mapping: feature difference Ξ = x - x' β braid word.
K(x,x') = β¨0|U_Ξ¦(x) U_Ξ¦(x')β |0β© where U_Ξ¦ encodes braid.
"""
function feature_diff_to_braid(x::Vector{Float64}, xβ²::Vector{Float64},
n_strands::Int; epsilon::Float64=0.5)::BraidWord
Ξ = x - xβ²
generators = Int[]
edge_indices = Int[]
gens_map = heavy_hex_braid_generators(n_strands)
n_gens = length(gens_map)
for (i, Ξ΄) in enumerate(Ξ)
if abs(Ξ΄) < epsilon
continue
end
gen_idx = (i - 1) % n_gens + 1
edge = gens_map[gen_idx]
edge_idx = HERON_EDGE_INDEX[edge]
sign = Ξ΄ > 0 ? 1 : -1
repeats = min(max(1, Int(round(abs(Ξ΄) * 2))), 3)
for _ in 1:repeats
push!(generators, sign * gen_idx)
push!(edge_indices, edge_idx)
end
end
if isempty(generators)
return BraidWord(n_strands)
end
BraidWord(generators, edge_indices, n_strands)
end
# βββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββ
# Batch Operations
# βββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββ
function pairwise_braid_words(X::Matrix{Float64}, n_strands::Int;
epsilon::Float64=0.5)::Matrix{BraidWord}
n_samples = size(X, 2)
braids = Matrix{BraidWord}(undef, n_samples, n_samples)
for i in 1:n_samples, j in 1:n_samples
braids[i,j] = feature_diff_to_braid(X[:,i], X[:,j], n_strands; epsilon=epsilon)
end
return braids
end
end # module TDABraidMap
|