Skip to content

6G-GOALS/semantic-dict-sheaf

 
 

Repository files navigation

Learning Network Sheaves for AI-native Semantic Communication

arXiv License


Tip

Recent advances in AI call for a paradigm shift from bit-centric communication to goal- and semantics-oriented architectures, paving the way for AI-native 6G networks. In this context, we address a key open challenge: enabling heterogeneous AI agents to exchange compressed latent-space representations while mitigating semantic noise and preserving task-relevant meaning. We cast this challenge as learning both the communication topology and the alignment maps that govern information exchange among agents, yielding a learned network sheaf equipped with orthogonal maps. This learning process is further supported by a semantic denoising and compression module that constructs a shared global semantic space and derives sparse, structured representations of each agent’s latent space. This corresponds to a nonconvex dictionary learning problem solved iteratively with closed-form updates. Experiments with multiple AI agents pre-trained on real image data show that the semantic denoising and compression facilitates AI agents alignment and the extraction of semantic clusters, while preserving high accuracy in downstream task. The resulting communication network provides new insights about semantic heterogeneity across agents, highlighting the interpretability of our methodology.

Dependencies

Using pip package manager

It is highly recommended to create a Python virtual environment before installing dependencies. In a terminal, navigate to the root folder and run:

python -m venv <venv_name>

Activate the environment:

  • On macOS/Linux:
source <venv_name>/bin/activate
  • On Windows:
<venv_name>\Scripts\activate

Once the virtual environment is active, install the dependencies:

pip install -r requirements.txt

You're ready to go! 🚀

Using uv package manager (Highly Recommended)

uv is a modern Python package manager that is significantly faster than pip.

Install uv

To install uv, follow the instructions from the official installation guide.

Set up the environment and install dependencies

Run the following command in the root folder:

uv sync

This will automatically create a virtual environment (if none exists) and install all dependencies.

You're ready to go! 🚀

Citation

If you find this code useful for your research, please consider citing the following paper:

@misc{grimaldi2025learningnetworksheavesainative,
      title={Learning Network Sheaves for AI-native Semantic Communication}, 
      author={Enrico Grimaldi and Mario Edoardo Pandolfo and Gabriele D'Acunto and Sergio Barbarossa and Paolo Di Lorenzo},
      year={2025},
      eprint={2512.03248},
      archivePrefix={arXiv},
      primaryClass={cs.MA},
      url={https://arxiv.org/abs/2512.03248}, 
}

Authors

Used Technologies

Python PyTorch SciPy NumPy

About

This repository implements a dictionary-driven, sheaf-based semantic communication framework for heterogeneous AI agents, learning both the communication topology and sheaf maps for latent-space alignment, supported by a sparse dictionary–based semantic denoising and compression module that builds a shared global semantic space.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

 
 
 

Contributors

Languages

  • Python 100.0%