Index

_

__init__() (structured_stochasticity.evaluation.AggregatedResult method) (structured_stochasticity.evaluation.EvaluationResult method) (structured_stochasticity.evaluation.Evaluator method) (structured_stochasticity.evaluation.TrajectoryAggregator method) (structured_stochasticity.experiment.Experiment method) (structured_stochasticity.experiment.ExperimentConfig method) (structured_stochasticity.hooks.HiddenStateHook method) (structured_stochasticity.hooks.HookConfig method) (structured_stochasticity.hooks.NoisyInferenceWrapper method) (structured_stochasticity.injection.AnnealedNoiseInjector method) (structured_stochasticity.injection.GaussianNoiseInjector method) (structured_stochasticity.injection.InjectionState method) (structured_stochasticity.injection.LayerSelectiveInjector method) (structured_stochasticity.injection.NoiseInjector method) (structured_stochasticity.injection.OncePerGenerationInjector method) (structured_stochasticity.injection.UniformNoiseInjector method) (structured_stochasticity.tasks.ArithmeticSequence method) (structured_stochasticity.tasks.TaskInstance method) (structured_stochasticity.tasks.TaskResult method) (structured_stochasticity.tasks.TowerOfHanoi method)

A

accuracy_by_complexity (structured_stochasticity.evaluation.EvaluationResult attribute) advance() (structured_stochasticity.injection.InjectionState method) aggregate() (structured_stochasticity.evaluation.TrajectoryAggregator method) AggregatedResult (class in structured_stochasticity.evaluation)

agreement_rate (structured_stochasticity.evaluation.AggregatedResult attribute) all_results (structured_stochasticity.evaluation.EvaluationResult attribute) AnnealedNoiseInjector (class in structured_stochasticity.injection) any_correct (structured_stochasticity.evaluation.AggregatedResult attribute) ArithmeticSequence (class in structured_stochasticity.tasks)

C

compare_k_scaling() (structured_stochasticity.evaluation.Evaluator method) COMPARISONS (structured_stochasticity.tasks.LogicalDeduction attribute) complexity (structured_stochasticity.tasks.TaskInstance attribute) (structured_stochasticity.tasks.TaskResult attribute) complexity_range (structured_stochasticity.experiment.ExperimentConfig attribute)

create() (structured_stochasticity.tasks.TaskFactory class method) create_injector() (in module structured_stochasticity.injection) current_layer (structured_stochasticity.injection.LayerSelectiveInjector attribute) current_scale (structured_stochasticity.injection.AnnealedNoiseInjector property) (structured_stochasticity.injection.LayerSelectiveInjector property)

D

device (structured_stochasticity.experiment.ExperimentConfig attribute)

E

enabled (structured_stochasticity.hooks.HookConfig attribute) error_message (structured_stochasticity.tasks.TaskResult attribute) evaluate() (structured_stochasticity.evaluation.Evaluator method) EvaluationResult (class in structured_stochasticity.evaluation)

Evaluator (class in structured_stochasticity.evaluation) Experiment (class in structured_stochasticity.experiment) experiment_name (structured_stochasticity.experiment.ExperimentConfig attribute) ExperimentConfig (class in structured_stochasticity.experiment)

F

from_yaml() (structured_stochasticity.experiment.ExperimentConfig class method)

G

GaussianNoiseInjector (class in structured_stochasticity.injection) generate() (structured_stochasticity.hooks.NoisyInferenceWrapper method) generate_batch() (structured_stochasticity.tasks.Task method) generate_instance() (structured_stochasticity.tasks.ArithmeticSequence method) (structured_stochasticity.tasks.LogicalDeduction method) (structured_stochasticity.tasks.Task method) (structured_stochasticity.tasks.TowerOfHanoi method)

generate_responses() (structured_stochasticity.experiment.Experiment method) generate_trajectories() (structured_stochasticity.hooks.NoisyInferenceWrapper method) generate_trajectories_decoded() (structured_stochasticity.hooks.NoisyInferenceWrapper method) get_stats() (structured_stochasticity.hooks.NoisyInferenceWrapper method) ground_truth (structured_stochasticity.tasks.TaskInstance attribute)

H

handle (structured_stochasticity.hooks.HiddenStateHook attribute) HiddenStateHook (class in structured_stochasticity.hooks)

HookConfig (class in structured_stochasticity.hooks) hooks (structured_stochasticity.hooks.NoisyInferenceWrapper attribute)

I

individual_results (structured_stochasticity.evaluation.AggregatedResult attribute) inject() (structured_stochasticity.injection.NoiseInjector method) injection_layers (structured_stochasticity.experiment.ExperimentConfig attribute) injection_mode (structured_stochasticity.experiment.ExperimentConfig attribute)

injection_point (structured_stochasticity.hooks.HookConfig attribute) InjectionState (class in structured_stochasticity.injection) injectors (structured_stochasticity.hooks.NoisyInferenceWrapper attribute) is_correct (structured_stochasticity.evaluation.AggregatedResult attribute) (structured_stochasticity.tasks.TaskResult attribute)

K

k_trajectories (structured_stochasticity.evaluation.AggregatedResult attribute) (structured_stochasticity.evaluation.EvaluationResult attribute)

k_values (structured_stochasticity.experiment.ExperimentConfig attribute)

L

last_input_norm (structured_stochasticity.hooks.HiddenStateHook attribute) last_output_norm (structured_stochasticity.hooks.HiddenStateHook attribute) LAYER_PATTERNS (structured_stochasticity.hooks.NoisyInferenceWrapper attribute)

layers (structured_stochasticity.hooks.HookConfig attribute) LayerSelectiveInjector (class in structured_stochasticity.injection) list_tasks() (structured_stochasticity.tasks.TaskFactory class method) LogicalDeduction (class in structured_stochasticity.tasks)

M

main() (in module structured_stochasticity.experiment) max_new_tokens (structured_stochasticity.experiment.ExperimentConfig attribute) max_solved_complexity (structured_stochasticity.evaluation.EvaluationResult attribute) metadata (structured_stochasticity.evaluation.AggregatedResult attribute) (structured_stochasticity.tasks.TaskInstance attribute) (structured_stochasticity.tasks.TaskResult attribute)

model_name (structured_stochasticity.experiment.ExperimentConfig attribute) module structured_stochasticity.evaluation structured_stochasticity.experiment structured_stochasticity.hooks structured_stochasticity.injection structured_stochasticity.tasks

N

name (structured_stochasticity.tasks.ArithmeticSequence attribute) (structured_stochasticity.tasks.LogicalDeduction attribute) (structured_stochasticity.tasks.Task attribute) (structured_stochasticity.tasks.TowerOfHanoi attribute) NAMES (structured_stochasticity.tasks.LogicalDeduction attribute)

noise_scale (structured_stochasticity.experiment.ExperimentConfig attribute) noise_strategy (structured_stochasticity.experiment.ExperimentConfig attribute) NoiseInjector (class in structured_stochasticity.injection) NoisyInferenceWrapper (class in structured_stochasticity.hooks) num_correct (structured_stochasticity.evaluation.AggregatedResult attribute)

O

OncePerGenerationInjector (class in structured_stochasticity.injection)

oracle_accuracy_by_complexity (structured_stochasticity.evaluation.EvaluationResult attribute) output_dir (structured_stochasticity.experiment.ExperimentConfig attribute)

P

print_summary() (structured_stochasticity.experiment.Experiment method)

prompt (structured_stochasticity.tasks.TaskInstance attribute)

R

register() (structured_stochasticity.hooks.HiddenStateHook method) register_hooks() (structured_stochasticity.hooks.NoisyInferenceWrapper method) remove() (structured_stochasticity.hooks.HiddenStateHook method) remove_hooks() (structured_stochasticity.hooks.NoisyInferenceWrapper method) reset() (structured_stochasticity.injection.InjectionState method) (structured_stochasticity.injection.NoiseInjector method) (structured_stochasticity.injection.OncePerGenerationInjector method)

reset_injectors() (structured_stochasticity.hooks.NoisyInferenceWrapper method) reset_stats() (structured_stochasticity.hooks.HiddenStateHook method) response (structured_stochasticity.tasks.TaskResult attribute) run() (structured_stochasticity.experiment.Experiment method) run_quick_experiment() (in module structured_stochasticity.experiment)

S

sample() (structured_stochasticity.injection.AnnealedNoiseInjector method) (structured_stochasticity.injection.GaussianNoiseInjector method) (structured_stochasticity.injection.LayerSelectiveInjector method) (structured_stochasticity.injection.NoiseInjector method) (structured_stochasticity.injection.OncePerGenerationInjector method) (structured_stochasticity.injection.UniformNoiseInjector method) save_results() (structured_stochasticity.experiment.Experiment method) selected_response (structured_stochasticity.evaluation.AggregatedResult attribute) selection_method (structured_stochasticity.evaluation.AggregatedResult attribute) (structured_stochasticity.evaluation.EvaluationResult attribute) (structured_stochasticity.experiment.ExperimentConfig attribute) set_enabled() (structured_stochasticity.hooks.NoisyInferenceWrapper method)

set_layer() (structured_stochasticity.injection.LayerSelectiveInjector method) setup() (structured_stochasticity.experiment.Experiment method) step (structured_stochasticity.injection.InjectionState attribute) structured_stochasticity.evaluation module structured_stochasticity.experiment module structured_stochasticity.hooks module structured_stochasticity.injection module structured_stochasticity.tasks module

T

Task (class in structured_stochasticity.tasks) task_name (structured_stochasticity.evaluation.EvaluationResult attribute) (structured_stochasticity.experiment.ExperimentConfig attribute) TaskFactory (class in structured_stochasticity.tasks) TaskInstance (class in structured_stochasticity.tasks) TaskResult (class in structured_stochasticity.tasks)

TASKS (structured_stochasticity.tasks.TaskFactory attribute) to_dict() (structured_stochasticity.experiment.ExperimentConfig method) torch_dtype (structured_stochasticity.experiment.ExperimentConfig attribute) total_steps (structured_stochasticity.injection.InjectionState attribute) TowerOfHanoi (class in structured_stochasticity.tasks) TrajectoryAggregator (class in structured_stochasticity.evaluation) trials_per_complexity (structured_stochasticity.experiment.ExperimentConfig attribute)

U

UniformNoiseInjector (class in structured_stochasticity.injection)

V

verify_solution() (structured_stochasticity.tasks.ArithmeticSequence method) (structured_stochasticity.tasks.LogicalDeduction method) (structured_stochasticity.tasks.Task method) (structured_stochasticity.tasks.TowerOfHanoi method)