Initial commit: ER-TP-DGP research prototype

Event-Reified Temporal Provenance Dual-Granularity Prompting for
LLM-based APT detection on DARPA provenance datasets.

Includes phase 0-14 method spec, IR/graph/metapath/trimming/prompt
modules, scripts for THEIA candidate universe, landmark CSG construction,
hybrid prompting, and LLM inference. Excludes data/, reports/, and
local LLM config from version control.
This commit is contained in:
BattleTag
2026-05-15 16:53:57 +08:00
commit b86ae87b75
88 changed files with 18570 additions and 0 deletions

35
pyproject.toml Normal file
View File

@@ -0,0 +1,35 @@
[project]
name = "er-tp-dgp"
version = "0.1.0"
description = "Event-Reified Temporal Provenance Dual-Granularity Prompting for LLM-based APT detection"
requires-python = ">=3.10"
readme = "README.md"
license = { text = "Research prototype" }
authors = [{ name = "ER-TP-DGP collaborators" }]
dependencies = ["PyYAML>=6.0"]
[project.optional-dependencies]
dev = ["pytest>=7.0"]
local = [
"torch>=2.3",
"transformers>=4.45",
"peft>=0.12",
"accelerate>=0.34",
"bitsandbytes>=0.43",
"datasets>=2.20",
"numpy>=1.26",
]
embed = [
"sentence-transformers>=3.0",
"numpy>=1.26",
]
eval = [
"scikit-learn>=1.4",
]
[tool.pytest.ini_options]
pythonpath = [".", "src"]
testpaths = ["tests"]
[tool.ruff]
line-length = 100