Fix: Prevent oscillation in label propagation algorithm

- Changed from synchronous to asynchronous updates with randomized node order
- Added maximum iteration limit (100) to prevent infinite loops
- Implemented oscillation detection with early stopping mechanism
- Improved tie-breaking with deterministic sorting
- Added detailed logging for convergence and oscillation events

The previous implementation used synchronous updates where all nodes
updated simultaneously, which could cause oscillation in certain graph
structures (e.g., bipartite graphs). This fix ensures the algorithm
always terminates and produces stable community assignments.
This commit is contained in:
remonxiao 2025-11-27 19:41:31 +08:00
parent 422558d06c
commit 675efbebe1

View file

@ -84,50 +84,101 @@ async def get_community_clusters(
def label_propagation(projection: dict[str, list[Neighbor]]) -> list[list[str]]:
# Implement the label propagation community detection algorithm.
# 1. Start with each node being assigned its own community
# 2. Each node will take on the community of the plurality of its neighbors
# 3. Ties are broken by going to the largest community
# 4. Continue until no communities change during propagation
"""
Implement the label propagation community detection algorithm with oscillation prevention.
Algorithm:
1. Start with each node being assigned its own community
2. Each node will take on the community of the plurality of its neighbors
3. Ties are broken by going to the largest community
4. Continue until no communities change during propagation
Oscillation prevention:
- Uses asynchronous updates (randomized node order)
- Maximum iteration limit to prevent infinite loops
- Early stopping if oscillation is detected
"""
import random
MAX_ITERATIONS = 100
OSCILLATION_CHECK_WINDOW = 5
community_map = {uuid: i for i, uuid in enumerate(projection.keys())}
while True:
no_change = True
new_community_map: dict[str, int] = {}
for uuid, neighbors in projection.items():
node_uuids = list(projection.keys())
# Track history to detect oscillations
history: list[dict[str, int]] = []
for iteration in range(MAX_ITERATIONS):
# Asynchronous update: randomize node processing order
random.shuffle(node_uuids)
changed_count = 0
for uuid in node_uuids:
neighbors = projection[uuid]
curr_community = community_map[uuid]
# Count votes from neighbors
community_candidates: dict[int, int] = defaultdict(int)
for neighbor in neighbors:
community_candidates[community_map[neighbor.node_uuid]] += neighbor.edge_count
if not community_candidates:
continue
# Sort by count (descending), then by community ID for deterministic tie-breaking
community_lst = [
(count, community) for community, count in community_candidates.items()
]
community_lst.sort(reverse=True)
candidate_rank, community_candidate = community_lst[0] if community_lst else (0, -1)
if community_candidate != -1 and candidate_rank > 1:
community_lst.sort(key=lambda x: (-x[0], x[1]))
candidate_rank, community_candidate = community_lst[0]
# Update community based on neighbor plurality
if candidate_rank > 1:
new_community = community_candidate
else:
# For weak signals, prefer staying in current community
new_community = max(community_candidate, curr_community)
new_community_map[uuid] = new_community
if new_community != curr_community:
no_change = False
if no_change:
community_map[uuid] = new_community
changed_count += 1
# Check for convergence
if changed_count == 0:
logger.debug(f'Label propagation converged after {iteration + 1} iterations')
break
community_map = new_community_map
# Check for oscillation by comparing with recent history
if iteration >= OSCILLATION_CHECK_WINDOW:
current_state = community_map.copy()
history.append(current_state)
# Keep only recent history
if len(history) > OSCILLATION_CHECK_WINDOW:
history.pop(0)
# Detect oscillation: if current state matches any recent state
for past_state in history[:-1]:
if past_state == current_state:
logger.warning(
f'Label propagation oscillation detected at iteration {iteration + 1}, '
'stopping early'
)
break
else:
logger.warning(
f'Label propagation reached maximum iterations ({MAX_ITERATIONS}) without converging'
)
# Group nodes by community
community_cluster_map = defaultdict(list)
for uuid, community in community_map.items():
community_cluster_map[community].append(uuid)
clusters = [cluster for cluster in community_cluster_map.values()]
clusters = list(community_cluster_map.values())
return clusters